BZOJ4407: 于神之怒加强版(莫比乌斯反演 线性筛)

Description

给下N,M,K.求

感觉好迷茫啊，很多变换看的一脸懵逼却又不知道去哪里学。一道题做一上午也是没谁了，，

首先按照套路反演化到最后应该是这个式子

$$ans = \sum_{d = 1}^n d^k \sum_{i = 1}^{\frac{n}{d}} \frac{n}{di} \frac{m}{di} \mu(i)$$

这样就可以$O(n)$计算

继续往下推，考虑$\frac{n}{di} \frac{m}{di}$对答案的贡献

设$T = id$

$ans = \sum_{T = 1}^n \frac{n}{T} \frac{m}{T} \sum_{d \mid T} ^ T d^k \mu(\frac{T}{d})$

后面那一坨是狄利克雷卷积的形式，显然是积性函数，可以直接筛

然后我在这里懵了一个小时，，

设$H(T) = \sum_{d \mid T} ^ T d^k \mu(\frac{T}{d})$

那么当$T = p^a$式，上面的式子中只有$\frac{T}{d} = 1$或$\frac{T}{d} = p$式，$\mu(\frac{T}{d})$才不为$0$

那么把式子展开$H(p^{a + 1}) = H(p^a) * (p^k)$

// luogu-judger-enable-o2
#include<cstdio>
#include<algorithm>
#define LL long long
using namespace std;
const int MAXN =  * 1e6 + , mod = 1e9 + ;
inline int read() {
    char c = getchar(); int x = , f = ;
    while(c < '' || c > '') {if(c == '-') f = -; c = getchar();}
    while(c >= '' && c <= '') x = x *  + c - '', c = getchar();
    return x * f;
}
int T, K;
int prime[MAXN], vis[MAXN], tot, mu[MAXN];
LL H[MAXN], low[MAXN];
LL fastpow(LL a, LL p) {
    LL base = ;
    while(p) {
        if(p & ) base = (base * a) % mod;
        a = (a * a) % mod; p >>= ;
    }
    return base;
}
void GetH(int N) {
    vis[] = H[] = mu[] = low[] = ;
    for(int i = ; i <= N; i++) {
        if(!vis[i]) prime[++tot] = i, mu[i] = -, H[i] = (- + fastpow(i, K) + mod) % mod, low[i] = i;
        for(int j = ; j <= tot && i * prime[j] <= N; j++) {
            vis[i * prime[j]] = ;
            if(!(i % prime[j])) {
                mu[i * prime[j]] = ; low[i * prime[j]] = (low[i] * prime[j]) % mod;
                if(low[i] == i)
                    //H[i * prime[j]] = (H[i] + fastpow((i * prime[j]), K)) % mod;
                    H[i * prime[j]] = H[i] * (fastpow(prime[j], K)) % mod;
                else H[i * prime[j]] = H[i / low[i]] * H[prime[j] * low[i]] % mod;
                break;
            }
            mu[i * prime[j]] = mu[i] * mu[prime[j]] % mod;
            H[i * prime[j]] = H[i] * H[prime[j]] % mod;
            low[i * prime[j]] = prime[j] % mod;
        }
    }
    for(int i = ; i <= N; i++) H[i] = (H[i] + H[i - ] + mod) % mod;
}
int main() {
    T = read(); K = read();
    GetH();
    while(T--) {
        int N = read(), M = read(), last;
        LL ans = ;
        if(N > M) swap(N, M);
        for(int T = ; T <= N; T = last + ) {
            last = min(N / (N / T), M / (M / T));
            ans = (ans + (1ll * (N / T) * (M / T) % mod) * (H[last] - H[T - ] + mod)) % mod;
        }
        printf("%lld\n", ans % mod);
    }
    return ;
}
/*
2 5000000
7 8
123 456
4999999 5000000
*/