bzoj 3994

d(x) x 的约数个数,给定 N,M ,求

i=1Nj=1Md(ij)

Input

输入文件包含多组测试数据。
第一行,一个整数 T ,表示测试数据的组数。
接下来的 T 行,每行两个整数 N,M

Output

T 行,每行一个整数,表示你所求的答案。

Sample Input

2
7 4
5 6

Sample Output

110
121

HINT

1N,M50000
1T50000

这题有个很屌的结论:

i=1Nj=1Md(ij)=i=1Nj=1MNiMj[gcd(i,j)=1]

根据 PoPoQQQ博客所说的,我们可以先证明这个式子的成立:
d(nm)=injm[gcd(i,j)=1]

我们可以证明一下:我们对每个质数 p 单独算贡献,设 n=n×pk1 m=m×pk2 。那么,该质数 p 对答案的贡献显然为 k1+k2+1 。于是我们考虑
d(nm)=injm[gcd(i,j)=1]
这个式子,发现 p 对之有贡献的数对 (i,j) 仍然是
(pk1,1),(pk11,1)(1,1)(1,pk21),(1,pk2)
k1+k2+1 个,因此得证。
代入得
n=1Nm=1Md(nm)=n=1Nm=1Minjm[gcd(i,j)=1]

我们转变枚举量,先枚举 i,j 就有
i=1Nj=1MNiMj[gcd(i,j)=1]

于是
i=1Nj=1MNiMj[gcd(i,j)=1]
这个式子我们可以上反演了。
反演化为
i=1Nj=1MNiMjgigjμ(g)

转而枚举 g ,于是就可得到
g=1Nμ(g)i=1Ngj=1MgNigMjg

再化一下就可得到
g=1Nμ(g)i=1NgNigj=1MgMjg

又有
Nab=Nab

于是我们发现 Ngi=1Nig 只与 Ng 有关,我们可以 O(nn) 预处理
fx=i=1xxi

有了这个后再化简
g=1Nμ(g)fNgfMg
就可在 O(n) 分段求了。皆大欢喜。

#include<iostream>
#include<cstdio>
#include<cstdlib>
using namespace std;

typedef long long ll;
#define maxn (50010)
int f[maxn],mu[maxn],prime[maxn],n,m,tot; bool exist[maxn];

inline int calc(int x) {
    int ret = 0;
    for (int i = 1,last;i <= x;i = last+1)
    {
        last = min(x,x/(x/i));
        ret += (x/i)*(last-i+1);
    }
    return ret;
}
inline void ready() {
    mu[1] = 1;
    for (int i = 2;i <= 50000;++i)
    {
        if (!exist[i]) { prime[++tot] = i; mu[i] = -1; }
        for (int j = 1;j <= tot&&prime[j]*i <= 50000;++j)
        {
            exist[i*prime[j]] = true;
            if (i % prime[j] == 0) { mu[i*prime[j]] = 0; break; }
            mu[i*prime[j]] = -mu[i];
        }
    }
    for (int i = 1;i <= 50000;++i) mu[i] += mu[i-1],f[i] = calc(i);
}

inline ll work() {
    if (n > m) swap(n,m);
    ll ret = 0;
    for (int i = 1,last;i <= n;i = last+1)
    {
        last = min(n,min(n/(n/i),m/(m/i)));
        ret += (ll)(mu[last]-mu[i-1])*((ll)f[n/i]*f[m/i]);
    }
    return ret;
}

int main() {
    freopen("3994.in","r",stdin);
    freopen("3994.out","w",stdout);
    ready();
    int T; scanf("%d",&T);
    while (T--) scanf("%d %d",&n,&m),printf("%lld\n",work());
    fclose(stdin); fclose(stdout);
    return 0;
}

你可能感兴趣的:(bzoj 3994)