UVA 11246 - K-Multiple Free set (数论/容斥)

题意

给定ｎ和ｋ，求一个包含元素只有1-n的集合，集合中不存在一个元素是另一个元素的ｋ倍，求集合的最大大小

思路

首先，若a%k!=0,a*k的数都是要删掉的，但是a*k*k可以存在,a*k*k*k又应该删掉，以此类推就是因子有ｋ的奇数次幂的删掉，偶数次加回来

代码

#include <bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof(a))
#define rep(i,a,b) for(int i=a;i<b;i++)
#define sc(a) scanf("%d",&a)
const int INF=0x3f3f3f3f;
const int maxn=1e6+50;
const int mod=1e9+7;
const double eps=1e-8;
#define pii pair<int,int>
typedef long long ll;
typedef unsigned int ui;
using namespace std;

int s[maxn];
int mins[maxn],maxs[maxn];

int bl(int n,int k){
    map<int,bool> ma;
    vector<int> ers;
    rep(i,1,n+1) {
        if(i%k==0){
            if(ma.find(i/k)==ma.end()) ma[i]=true;
            else ers.push_back(i);
        }else ma[i]=true;
    }
    for(int i=0;i<ers.size();i++){
        printf("%d ",ers[i]);
        if(i) printf("= =%d",ers[i]-ers[i-1]);
        puts("");
    }
    puts("");
    for(auto it=ma.begin();it!=ma.end();it++){
        printf("%d ",it->first);
    }
    puts("");
    return ma.size();
}

int solve(int n,int k){
    int ret=n;
    int flag=1;
    for(ll kp=k;kp<=n;kp*=k){
        flag=-flag;
        ret=ret+n*flag/(int)kp;
    }
    return ret;
}
int main()
{
#ifndef ONLINE_JUDGE
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
#endif
    int n,k,T; sc(T);
    while(T--){
        sc(n); sc(k);
        //cout<<"ans: "<<bl(n,k)<<" "<<solve(n,k)<<endl;
        printf("%d\n",solve(n,k));
    }
    return 0;
}