BZOJ3309 DZY loves Math

Problem

BZOJ

Solution

本来想写完DZY loves Math系列再统一发题解的，然后我就被第二题劝退了，所以就只能把这个计划暂时鸽搁下来了。

$$\sum _{i=1}^n\sum _{j=1}^{m}f(\gcd (a,b))$$

$$\sum _{d=1}^{n}f(d)\sum _{i=1}^n\sum _{j=1}^m[\gcd (i,j)=d]$$

$$\sum _{d=1}^{n}f(d)\sum _{i=1}^{n/d}\mu (i)\frac{n}{id}\frac{m}{id}$$

$$\sum _{id=1}^n\frac{n}{id}\frac{m}{id}\sum _{i|id}f(\frac{id}{i})\mu (i)$$

令$g(n)={\sum }_{d|n}f(\frac{n}{i})\mu (i)$，打表找规律，发现若$n={\prod }_{i=1}^k{p}_i^{a_i}$，当$a_i$均相等时$g(n)=(-1{)}^{k+1}$，否则$g(n)=0$。

这个证明也不难，考虑莫比乌斯函数只有在质数的指数小于等于1时有值。

• 若存在$a_i<a_j$，那么我们$f(d)$中的$d$取不取$p_i$都一样，而莫比乌斯函数的取值却恰好相反，所以抵消了，恒等于0
• 若不存在，如果$\mu$中的值取了$k$个质数，$f(d)=a-1$，其它$2^k-1$中情况，$f(d)=a$。由于奇偶性对称，所以最后的值应该是$(-1{)}^k(a-1)+(-1{)}^{k+1}a=(-1{)}^{k+1}$

关于实现上的trick：不需要记录mn和mx，只需要判定mn和除去mn的贡献之后是否一样即可。

Code

#include <algorithm>
#include <cstdio>
using namespace std;
typedef long long ll;
const int maxn=10000010;
template <typename Tp> inline int getmin(Tp &x,Tp y){return y<x?x=y,1:0;}
template <typename Tp> inline int getmax(Tp &x,Tp y){return y>x?x=y,1:0;}
template <typename Tp> inline void read(Tp &x)
{
    x=0;int f=0;char ch=getchar();
    while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
    if(ch=='-') f=1,ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    if(f) x=-x;
}
int z,n,m,tot,pri[maxn],d[maxn],mn[maxn],val[maxn];
ll ans;
void init()
{
	for(int i=2;i<=10000000;i++)
	{
		if(!d[i]) pri[++tot]=i,d[i]=mn[i]=1,val[i]=1;
		for(int j=1;j<=tot&&i*pri[j]<=10000000;j++)
		{
			int r=i*pri[j];
			if(i%pri[j]==0)
			{
				d[r]=d[i];mn[r]=mn[i]+1;
				if(d[i]==1) val[r]=1;
				else val[r]=(mn[r]==mn[d[r]]?-val[d[r]]:0);
				break;
			}
			d[r]=i;mn[r]=1;
			val[r]=(mn[i]==1?-val[i]:0);
		}
	}
	for(int i=2;i<=10000000;i++) val[i]+=val[i-1];
}
int main()
{
	init();
	read(z);
	while(z--)
	{
		read(n);read(m);ans=0ll;
		if(n>m) swap(n,m);
		for(int i=1,j;i<=n;i=j+1)
		{
			j=min(n/(n/i),m/(m/i));
			ans+=(ll)(n/i)*(m/i)*(val[j]-val[i-1]);
		}
		printf("%lld\n",ans);
	}
	return 0;
}