poj 2773 容斥原理求第k個與m互質的數

二分答案，然後用容斥原理求1-mid中與m互質的個數

注意二分的上界

View Code

#include<stdio.h>
#include<vector>
using namespace std;
int m,n;
__int64 solve(int r,__int64 n){
    vector<int> p;
int i;
for(i=2;i*i<=r;i++){
if(r%i==0){
            p.push_back(i);
while(r%i==0) r/=i;
        }
    }
if(r>1) p.push_back(r);
    __int64 sum=0;
for(__int64 num=1;num<(1<<p.size());num++){
        __int64 mult=1,ones=0;
for(i=0;i<p.size();i++){
if(num&(1<<i)){
                ones++;
                mult*=p[i];
            }
        }
if(ones%2) sum+=n/mult;
else sum-=n/mult;
    }
return n-sum;
}
int main(){
int k,i,j;
while(scanf("%d%d",&m,&k)!=EOF){
        __int64 l=0,r=((__int64)1<<62);
        __int64 mid;
        __int64 ans=0;
while(l<=r){
             mid=(l+r)>>1;
            __int64 tmp=solve(m,mid);
if(tmp>=k){
                r=mid-1;
if(tmp==k) ans=mid;
            }
else l=mid+1;
        }
        printf("%I64d\n",ans);
    }
return 0;
}

還有一種非常暴力的方法：只求1-m中與m互質的數，隨後的互質的數就迴圈下去了

#include<stdio.h>
int p[1000010];
int gcd(int a,int b)
{
return b==0 ? a : gcd(b,a%b);
}
int main()
{
int m,k,i,j;
while(scanf("%d%d",&m,&k)!=EOF)
    {
int tot=0;
for(i=1;i<=m;i++)
        {
if(gcd(i,m)==1)
            {    
                p[++tot]=i;
            }
        }
int ans;
int tmp=k/tot;
int tmp2=k%tot;
if(tmp2==0) ans=(tmp-1)*m+p[tot];
else ans=m*(tmp)+p[tmp2];
        printf("%d\n",ans);
    }
}