URAL 1091 Tmutarakan Exams(容斥原理)

题目链接：
URAL 1091 Tmutarakan Exams
题意：
求构造$k$个不同的数，每个数不超过$s$，且这些数的最大公约数大于1的方案数。方案数大于$10000$时，直接输出$10000$。
数据范围：$2 \leq K \leq S \leq50$
分析：
我们只需要考虑最小非1公约数情况。$假设最小非1公约数为p$，那么在这$s$个数内，有$\frac{s}{p}$个数是$p$的倍数，从这些数中任意选择$k$个都是可以的，有$C_{\frac{s}{p}}^{k}$种选择。
但是这样会有重复，比如当$k=3$时：
在计算最小非1公约数为$2$时将序列：$4，8，12$计算了一次。
在计算最小非1公约数为 4 <script type="math/tex" id="MathJax-Element-1634">4</script>时也将上述序列计算了一次。
所以需要从大到小去掉重复的。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <climits>
#include <cmath>
#include <ctime>
#include <cassert>
#define IOS ios_base::sync_with_stdio(0); cin.tie(0);
using namespace std;
typedef long long ll;
const int MAX_N = 55;

ll C[MAX_N][MAX_N];

ll init()
{
    ll Max = 1;
    for(int i = 1; i < MAX_N; ++i) {
        C[i][0] = C[i][i] = 1;
        for(int j = 1; j < i; ++j) {
            C[i][j] = C[i - 1][j] + C[i - 1][j - 1];
        }
    }
}

ll ans[MAX_N];

ll solve(int k, int s)
{
    ll res = 0;
    memset(ans, 0, sizeof(ans));
    int total = 0;
    for(int i = 2; s / i >= k; ++i) {
        ans[i] = C[s / i][k];
        total = i;
    }
    for(int i = total; i >= 2; --i) {
        for(int j = 2; j < i; ++j) {
            if(i % j == 0) ans[j] -= ans[i];
        }
    }
    for(int i = 2; i <= total; ++i) {
        res += ans[i];
    }
    if(res > 10000) res = 10000;
    return res;
}

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