nyoj 第k个互质数（二分+容斥原理）

第k个互质数

描述

两个数的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

思路：

首先要知道，

    在[1,m]之间与m互质的数的个数=[1,m]之间的总个数-[1,m]之间与n不互质的数的个数

所以，要先对m进行质因数分解，求出m有哪些质因数，然后二分枚举答案mid，用容斥求[1,mid]内与m互质的数有多少个，让其结果与k比较

判断的时候，[1,mid]之间与m互质的数的数量 = mid - （包含一个质因子的数的个数）+（包含2个质因子的数的个数）-（包含3个质因子的数的个数）+（包含4个质因子的数的个数）.....

代码:

#include<stdio.h>
#include<vector>
using namespace std;

vector<int>a;//a来保存m的质因子
int b[100];

void div(int m)
{
    for(int i=2;i*i<=m;i++)
    {
        if(m%i==0)
            a.push_back(i);
        while(m%i==0)
            m/=i;
    }
    if(m>1)
        a.push_back(m);
}

int cont(int n)
{
    int g=0,sum=n;
    b[++g]=1;
    for(int i=0;i<a.size();i++)
    {
        int t=g;
        for(int j=1;j<=g;j++)
        {
            b[++t]=-1*a[i]*b[j];//偶数个质因子为正，奇数个质因子为负
            sum+=n/b[t];//计算[1,n]之间有多少数含有因子b[t]
        }
        g=t;
    }
    return sum;
}
//二分枚举答案mid，计算[1,mid]内有多少个数与m互质，并将其结果与k比较
int Binary_search(int m,int k)
{
    int le=1,ri=1000000000,mid;
    while(le<=ri)
    {
        mid=(le+ri)>>1;
        if(cont(mid)>=k)
            ri=mid-1;
        else
            le=mid+1;
    }
    return le;
}

int main()
{
    int m,k;
    while(~scanf("%d%d",&m,&k))
    {
        a.clear();
        div(m);
        printf("%d\n",Binary_search(m,k));
    }
    return 0;
}