ACM学习历程—ZOJ 3868 GCD Expectation（莫比乌斯 || 容斥原理）

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <set>
#include <queue>
#include <vector>
#define LL long long
#define MOD 998244353

using namespace std;

const int maxN = 1e6+10;
int n, k, len;
int pow2[maxN], num[maxN];
int prime[maxN], u[maxN];
bool vis[maxN];

//莫比乌斯反演
//F(n)和f(n)是定义在非负整数集合上的两个函数
//并且满足条件F(n) = sum(f(d)) (d|n)
//那么f(n) = sum(u(d)*F(n/d)) (d|n)
//case 1: if d = 1, u(d) = 1
//case 2: if d = p1*p2...pk, (pi均为互异素数), u(d) = (-1)^k
//case 3: else, u(d) = 0;
//性质1: sum(u(d)) (d|n) = 1 (n == 1) or 0 (n > 1)
//性质2: sum(u(d)/d) (d|n) = phi(n)/n
//另一种形式:F(d) = sum(f(n)) (d|n) => f(d) = sum(u(n/d)*F(n)) (d|n)
//线性筛选求莫比乌斯反演函数代码
void mobius()
{
    memset(vis, false,sizeof(vis));
    u[1] = 1;
    int cnt = 0;
    for(int i = 2; i < maxN; i++)
    {
        if(!vis[i])
        {
            prime[cnt++] = i;
            u[i] = -1;
        }
        for(int j = 0; j < cnt && i*prime[j] < maxN; j++)
        {
            vis[i*prime[j]] = true;
            if(i%prime[j])
                u[i*prime[j]] = -u[i];
            else
            {
                u[i*prime[j]] = 0;
                break;
            }
        }
    }
}

void init()
{
    mobius();
    pow2[0] = 1;
    for (int i = 1; i < maxN; ++i)
        pow2[i] = 2*pow2[i-1]%MOD;
}

void input()
{
    int tmp;
    len = 0;
    scanf("%d%d", &n, &k);
    memset(num, 0, sizeof(num));
    for (int i = 0; i < n; ++i)
    {
        scanf("%d", &tmp);
        len = max(len, tmp);
        num[tmp]++;
    }
    for (int i = 1; i <= len; ++i)
    {
        for (int j = i*2; j <= len; j += i)
            num[i] += num[j];
    }
}

//快速幂m^n
LL quickPow(LL x, int n)
{
    if (n == 0)
        return 1;
    if (x == 0)
        return 0;
    LL a = 1;
    while (n)
    {
        a *= n&1 ? x : 1;
        a %= MOD;
        n >>= 1 ;
        x *= x;
        x %= MOD;
    }
    return a;
}

void work()
{
    int ans = 0, val;
    for (int i = 1; i <= len; ++i)
    {
        val = quickPow(i, k);
        for (int j = i; j <= len; j += i)
        {
            ans += ((LL)u[j/i]*val%MOD)*(pow2[num[j]]-1)%MOD;
            ans = (ans%MOD+MOD)%MOD;
        }
    }
    printf("%d\n", ans);
}

int main()
{
    //freopen("test.in", "r", stdin);
    init();
    int T;
    scanf("%d", &T);
    for (int times = 1; times <= T; ++times)
    {
        input();
        work();
    }
    return 0;
}