现在的位置: 首页 > 综合 > 正文

NYOJ 762 第k个互质数(二分 + 容斥)

2017年10月17日 ⁄ 综合 ⁄ 共 1458字 ⁄ 字号 评论关闭

第k个互质数

时间限制:1000 ms  |  内存限制:65535 KB
难度:4
描述
两个数的a,b的gcd为1,即a,b互质,现在给你一个数m,你知道与它互质的第k个数是多少吗?与m互质的数按照升序排列。
输入
输入m ,k (1<=m<=1000000;1<=k<=100000000)
输出
输出第k个数。
样例输入
10 1
10 2
10 3
样例输出
1
3
7

首先,根据欧几里得可知,gcd(b * t + a, b) = gcd(a, b)(t为任意整数),则如果a与b互质,则b * t + a与b也一定互质,如果a与b不互质,则b*t+a与b也一定不互质,所以与m互质的数对m取模具有周期性。可以先求出1~m之间有多少个数与m互质,然后根据周期性去求第k个与m互质的数。

#include <cstdio>

const int N = 1000005;
int a[N];

int gcd(int a, int b) {
    while(b) {
        int r = a % b;
        a = b;
        b = r;
    }
    return a;
}

int main() {
    int m, k;
    while(~scanf("%d%d", &m, &k)) {
        if(m == 1) {
            printf("%d\n", k);
            continue;
        }
        int num = 0;
        for(int i = 1; i < m; i++)
            if(gcd(m, i) == 1)
                a[num++] = i;
        int p = k / num;
        if(k % num == 0) p--;
        k %= num;
        if(k == 0) k = num;
        printf("%d\n", p * m + a[k-1]);
    }
    return 0;
}

用上面的方法可以在POJ上AC,可惜的是,在NYOJ上TLE了。所以要寻找一个更快的解决方案,也就是下面的二分+容斥。

首先对m进行质因数分解,求出m有哪些质因数,然后用容斥求[1, mid]内与m互质的数有多少个。

判断的时候,[1,mid]之间与m互质的数的数量 = mid - (包含一个质因子的数的个数)+ (包含2个质因子的书的个数)-(包含3个质因子的数的个数)+ (包含4个质因数的数的个数)……

#include <cstdio>
// 对n进行素因子分解, fac[0]记录因子个数;
int fac[20];
void Div(int n) {
    int k = 0;
    for(int i = 2; i * i <= n; ++i){
        if(n % i == 0) fac[++k] = i;
        while(n % i == 0) n /= i;
    }
    if(n > 1) fac[++k] = n;
    fac[0] = k;
}
// 计算[1, n]内与m互质的数的个数
int que[1<<10];
int Count(int n, int m) {
    int g = 0, sum = n;
    que[++g] = 1;
    for(int i = 1; i <= fac[0]; ++i){
        int t = g;
        for(int j = 1; j <= g; ++j){
            que[++t] = que[j] * fac[i] * -1;
            sum += n / que[t];
        }
        g = t;
    }
    return sum;
}
// 二分,二分枚举一个答案mid,计算[1, mid]内有多少个数与m互质,让答案与K比较;
int Binary_search(int m, int K){
    int l = 1, r = 2000000000, mid;
    while(l <= r){
        mid = (l + r) >> 1;
        if(Count(mid, m) >= K) r = mid - 1;
        else l = mid + 1;
    }
    return l;
}
int main()
{
    int m, K;
    while(scanf("%d%d", &m, &K) != EOF)
    {
        Div(m);
        int ans = Binary_search(m, K);
        printf("%d\n", ans);
    }
    return 0;
}

抱歉!评论已关闭.