HDU 6390 容斥原理

题意

传送门 HDU 6390

题解

若 $gcd(a,b)=1$，则有 $$\varphi (a)\varphi (b)=\varphi (ab),G_u(a,b)=1$$ 假设 $a=p^{e1},b=p^{e2}$，则有 $$G_u(a,b)=\frac{p^{e1+e2}\ast (p-1)/p}{[p^{e1}\ast (p-1)/p]\ast [p^{e2}\ast (p-1)/p]}=\frac{p}{p-1}$$ 那么虑 $gcd(a,b)$ 的质因子即可，可以得到 $$G_u(a,b)=\frac{gcd(a,b)}{\varphi [gcd(a,b)]}$$ 线性筛预处理出欧拉函数，然后对相同 $gcd(a,b)$ 的情况进行统计。

容斥原理求解的基本思想是若 $d|n,d|m$，则 $gcd(n,m)\ge d$，计算 $[1,n]\times [1,m]$ 内约数为 $d$ 的情况是容易的，即 $\lfloor n/d\rfloor \times \lfloor m/d\rfloor$；那么从最大可能的约数向最小可能约数枚举，每次减去 $gcd$ 为当前枚举的约数的倍数的情况，同时统计答案即可。时间复杂度 $O(Tnlogn)$。

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
using namespace std;
#define maxn 1000005
typedef long long ll;
int prime[maxn >> 1], phi[maxn];
ll inv[maxn], f[maxn];
bool used[maxn];

void sieve()
{
    int p = 0;
    phi[1] = 1;
    for (int i = 2; i < maxn; ++i)
    {
        if (!used[i])
        {
            prime[p++] = i;
            phi[i] = i - 1;
        }
        for (int j = 0; j < p && prime[j] * i < maxn; ++j)
        {
            int k = prime[j] * i;
            used[k] = 1;
            if (i % prime[j] == 0)
            {
                phi[k] = phi[i] * prime[j];
                break;
            }
            else
                phi[k] = phi[i] * (prime[j] - 1);
        }
    }
}

int main()
{
    sieve();
    int t, n, m, p;
    scanf("%d", &t);
    while (t--)
    {
        scanf("%d%d%d", &m, &n, &p);
        int limit = min(n, m);
        inv[1] = 1;
        for (int i = 2; i <= limit; ++i)
            inv[i] = (p - p / i) * inv[p % i] % p;
        ll res = 0;
        for (int i = limit; i >= 1; --i)
        {
            f[i] = (ll)(n / i) * (m / i);
            for (int j = (i << 1); j <= limit; j += i)
                f[i] -= f[j];
            res += f[i] % p * i % p * inv[phi[i]] % p;
        }
        printf("%lld\n", res % p);
    }
    return 0;
}