poj2773(容斥+二分)

链接：点击打开链接

题意：求第K个与N互素的数

代码：

#include <map>
#include <set>
#include <queue>
#include <string>
#include <math.h>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;
const long long INF=0x3f3f3f3f;
long long N,K;
vector<long long> G;
long long judge(long long mid){
    long long i,j,k,n,id,ans,tmp;
    n=G.size(),ans=0;
    for(i=1;i<(1<<n);i++){
        j=i,k=id=0,tmp=1;
        while(j){
            if(j&1){
                k++;
                tmp*=G[id];
            }
            id++;
            j>>=1;
        }
        if(k%2!=0)
        ans+=(mid/tmp);
        else
        ans-=(mid/tmp);
    }
    return mid-ans;
}
int main(){
    long long i,j,l,r,mid,ans;                  //其实就是一个区间内与N互素的数的个数
    while(scanf("%I64d%I64d",&N,&K)!=EOF){      //的变形，所以直接二分一下位置，判断一下
        G.clear();                              //区间与N互素的个数
        for(i=2;i<=N;i++){
            if(N%i==0){
                G.push_back(i);
                while(N%i==0)
                N/=i;
            }
        }
        if(N>1)
        G.push_back(N);
        l=1,r=INF;
        while(l<=r){
            mid=(l+r)/2;
            if(judge(mid)>=K){
                ans=mid;
                r=mid-1;
            }
            else
            l=mid+1;
        }
        printf("%I64d\n",ans);
    }
    return 0;
}