HDU 5514 容斥原理

题意

传送门 HDU 5514

题解

对于青蛙 $i$，每一步即在模 $m$ 意义上增加 $a_i$，那么当青蛙走回起点 $0$ 时，有 $$a_i\times step=0\mathrm{m}\mathrm{o}\mathrm{d}m$$ 显然最小的步数为 $m/gcd(a_i,m)$，那么步长为 $gcd(a_i,m)$。青蛙可能的步长只能是 $m$ 的约数，约数个数的上界为 $2\sqrt{m}$，这是一个宽松的上界，实际上 $10^9$ 内的数字其约数个数仅 $10^3$ 级别。考虑对约数进行处理。

设约数数组为 $D$，使之有序。枚举每一只青蛙，标记满足 $gcd(a_i,m)|D_j$ 的约数。从小到大枚举约数，以其为步长，根据等差数列求和公式可以得到被这个步长标记的石子编号的和；根据容斥原理，约数 $D_j$ 会被满足 $D_i|D_j,i<j$ 的约数重复统计，那么减去重复统计的部分，同时对满足 $D_j|D_k,j<k$ 的约数记录 $D_j$ 对其重复计数的数量。设 $m$ 的约数个数为 $d$，时间复杂度 $O(t{d}^2)$。

#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
typedef long long ll;
const int maxd = 1600, maxn = 10005;
int T, N, M, A[maxn], D[maxd], cnt[maxd];
bool used[maxd];

int gcd(int a, int b) { return b == 0 ? a : gcd(b, a % b); }

int main()
{
    scanf("%d", &T);
    for (int c = 1; c <= T; ++c)
    {
        scanf("%d%d", &N, &M);
        for (int i = 0; i < N; ++i)
            scanf("%d", A + i);
        memset(used, 0, sizeof(used));
        memset(cnt, 0, sizeof(cnt));
        int d = 0;
        for (int i = 1; i * i <= M; ++i)
            if (M % i == 0)
            {
                D[d++] = i;
                if (M / i != i)
                    D[d++] = M / i;
            }
        sort(D, D + d);
        for (int i = 0; i < N; ++i)
        {
            int g = gcd(M, A[i]);
            for (int j = 0; j < d; ++j)
                if (D[j] % g == 0)
                    used[j] = 1;
        }
        ll res = 0;
        for (int i = 0; i < d; ++i)
            if (used[i])
            {
                int k = D[i], x = (M - 1) / k, n = 1 - cnt[i];
                res += (ll)(x + 1) * x / 2 * k * n;
                for (int j = i + 1; j < d; ++j)
                    if (used[j] && D[j] % k == 0)
                        cnt[j] += n;
            }
        printf("Case #%d: %lld\n", c, res);
    }
    return 0;
}