NYOJ 762 第k个互质数（二分 + 容斥）

第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;
}