Codeforces 920G - List Of Integers

920G - List Of Integers

思路：容斥+二分

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))

vector<int>p;
ll ans=;
void dfs(int k,int cnt,ll s,ll r){
    if(k==p.size()){
        if(cnt&)ans-=r/s;
        else ans+=r/s;
        return ;
    }
    dfs(k+,cnt,s,r);
    dfs(k+,cnt+,s*p[k],r);
}
void init(ll n){
    p.clear();
    for(ll i=;i*i<=n;i++){
        if(n%i==){
            p.pb(i);
            while(n%i==)n/=i;
        }
    }
    if(n>)p.pb(n);
}
ll count(ll r){
    ans=;
    dfs(,,,r);
    return ans;
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie();
    int T;
    int x,p,k;
    cin>>T;
    while(T--){
        cin>>x>>p>>k;
        init(p);
        ll t=count(x);
        ll l=x,r=1e18,m=(l+r)>>;
        while(l<r){
            if(count(m)-t>=k)r=m;
            else l=m+;
            m=(l+r)>>;
        }
        cout<<m<<endl;
    }
    return ;
}