UVA - 11246 - K-Multiple Free set（容斥原理）

  

题意：给定n（1 <= n <= 10^9）和k（1 <= k <= 100），从1~n中选尽量多的整数，使的任意两个整数之间都不是k倍的关系。

容斥原理，（例如n = 20，k = 3，答案为16个数（1，9，2，18，4，12，5，15，7，8，10，11，13，14，16，17，19，20））

一、n个数内共有n / k组数成为连续的k倍关系

       （有1、3、9，2、6、18，4、12，5、15，7，8，10，11，13，14，16，17，19，20，共n - n / k = 14个）

二、但是互为k ^ 2倍的关系同时也被删除了，所以要加回来

       （有9，18，共n - n / k + n / (k^2) = 16个）

三、若还存在k ^ 3倍的关系同时也被多加了，所以要再减去

四、若还存在k ^ 4倍的关系同时也被多加了，所以要加回来

……

综上，结果为n - n / k + n / (k ^ 2) - n / (k ^ 3) + n / (k ^ 4)……即应用容斥原理

（无意发现memset对于初始化10 ^ 9等的大数组非常慢，交的时候没有删去即TLE）

#include<cstdio>
#include<cstring>
#include<cctype>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<sstream>
#include<iterator>
#include<algorithm>
#include<string>
#include<vector>
#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<list>
typedef long long ll;
typedef unsigned long long llu;
const int MAXN = 100 + 10;
const int MAXT = 1000000000 + 10;
const int INF = 0x7f7f7f7f;
const double pi = acos(-1.0);
const double EPS = 1e-6;
using namespace std;

int T;
ll n, k;

ll solve(){
    ll cur = 1, ans = n, tmp = n / k;
    while(tmp){
        ans -= cur * tmp;
        tmp /= k;
        cur = -cur;
    }
    return ans;
}

int main(){
    scanf("%d", &T);
    while(T--){
        scanf("%lld%lld", &n, &k);
        printf("%lld\n", solve());
    }
    return 0;
}