【刷题】BZOJ 3529 [Sdoi2014]数表

Description

有一张n×m的数表，其第i行第j列（1<=i<=n，1<=j<=m）的数值为能同时整除i和j的所有自然数之和。给定a，计算数表中不大于a的数之和。

Input

输入包含多组数据。
输入的第一行一个整数Q表示测试点内的数据组数，接下来Q行，每行三个整数n，m，a(|a| < =10^9)描述一组数据。

Output

对每组数据，输出一行一个整数，表示答案模2^31的值。

Sample Input

2
4 4 3
10 10 5

Sample Output

20
148

HINT

1<=n,m<=10^5 , 1<=Q<=2×10^4

Solution

莫比乌斯反演，膜拜PoPoQQQ
先不管a的限制，我们直接求数表中所有数之和\(ans'\)
设\(F(i)\)为i的约数和，\(g(i)\)为在限制n和m范围内gcd为i的对数的个数，那么

\[ans'=\sum_{i=1}^{min(n,m)}F(i)g(i) \]

\(g(i)\)的反演已经是老套路了，直接得出\(g(i)=\sum_{i|d}\mu(\frac{d}{n})\lfloor \frac{n}{d} \rfloor \lfloor \frac{m}{d} \rfloor\)
继续推

\[ans'=\sum_{i=1}^{min(n,m)}F(i)\sum_{i|d}\mu(\frac{d}{i})\lfloor \frac{n}{d} \rfloor \lfloor \frac{m}{d} \rfloor=\sum_{i=1}^{min(n,m)}F(i)\sum_{k=1}^{\lfloor \frac{min(n,m)}{i} \rfloor}\mu(k)\lfloor \frac{n}{ik} \rfloor \lfloor \frac{m}{ik} \rfloor \]

设\(T=ik\)

\[ans'=\sum_{T=1}^{min(n,m)}\lfloor \frac{n}{T} \rfloor \lfloor \frac{m}{T} \rfloor\sum_{k=1}^{\lfloor \frac{min(n,m)}{T} \rfloor}F(k)\mu(\lfloor \frac{T}{k} \rfloor) \]

前面部分整除分块，我们只要处理出后半部分的前缀和就行了
对于\(F(k)\)直接\(O(nlog_2n)\)暴力枚举约数和它的倍数
对于\(\mu\)这就不用说了吧。。。（看之前的文章）
好的，我们把改装之后的问题解决了，可是原问题呢？
回到真正的\(ans\)，因为有a的限制，所以那些\(F(i)\)大于a的是不能加贡献的
那么我们把询问按a排序，然后用树状数组维护\(F(k)\mu(\lfloor \frac{T}{k} \rfloor)\)的前缀和，只有当前询问的a大于扫描到的\(F(i)\)，才把\(F(i)\)能产生的贡献加入树状数组。这样就保证了不改加入的贡献不会被加入

#include<bits/stdc++.h>
#define ll long long
const int MAXT=20000+10,MAXN=100000+10,Mod=0x7fffffff;
int T,cnt,vis[MAXN],prime[MAXN],mu[MAXN],C[MAXN],ans[MAXT],limit;
struct question{
	int n,m,a;
	int id;
	inline bool operator < (const question &A) const{
		return a<A.a;
	};
};
question Q[MAXT];
struct node{
	int s;
	int id;
	inline bool operator < (const node &A) const{
		return s<A.s;
	};
};
node F[MAXN];
template<typename T> inline void read(T &x)
{
	T data=0,w=1;
	char ch=0;
	while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
	if(ch=='-')w=-1,ch=getchar();
	while(ch>='0'&&ch<='9')data=((T)data<<3)+((T)data<<1)+(ch^'0'),ch=getchar();
	x=data*w;
}
template<typename T> inline void write(T x,char c='\0')
{
	if(x<0)putchar('-'),x=-x;
	if(x>9)write(x/10);
	putchar(x%10+'0');
	if(c!='\0')putchar(c);
}
template<typename T> inline void chkmin(T &x,T y){x=(y<x?y:x);}
template<typename T> inline void chkmax(T &x,T y){x=(y>x?y:x);}
template<typename T> inline T min(T x,T y){return x<y?x:y;}
template<typename T> inline T max(T x,T y){return x>y?x:y;}
inline void init()
{
	for(register int i=1;i<=limit;++i)
		for(register int j=i;j<=limit;j+=i)F[j].s+=i;
	for(register int i=1;i<=limit;++i)F[i].id=i;
	std::sort(F+1,F+limit+1);
	memset(vis,1,sizeof(vis));
	vis[0]=vis[1]=0;
	mu[1]=1;
	for(register int i=2;i<=limit;++i)
	{
		if(vis[i])
		{
			prime[++cnt]=i;
			mu[i]=-1;
		}
		for(register int j=1;j<=cnt&&i*prime[j]<=limit;++j)
		{
			vis[i*prime[j]]=0;
			if(i%prime[j])mu[i*prime[j]]=-mu[i];
			else break;
		}
	}
}
inline int lowbit(int x)
{
	return x&(-x);
}
inline int sum(int x)
{
	int res=0;
	while(x)
	{
		res+=C[x];
		x-=lowbit(x);
	}
	return res;
}
inline void add(int x,int k)
{
	while(x<=limit)
	{
		C[x]+=k;
		x+=lowbit(x);
	}
}
inline int solve(int n,int m)
{
	int res=0;
	for(register int i=1;;)
	{
		if(i>min(n,m))break;
		int j=min(n/(n/i),m/(m/i));
		res+=(n/i)*(m/i)*(sum(j)-sum(i-1));
		i=j+1;
	}
	return res&Mod;
}
int main()
{
	read(T);
	for(register int i=1;i<=T;++i)
	{
		read(Q[i].n);read(Q[i].m);read(Q[i].a);
		Q[i].id=i;
		chkmax(limit,min(Q[i].n,Q[i].m));
	}
	init();
	std::sort(Q+1,Q+T+1);
	for(register int i=1,j=1;i<=T;++i)
	{
		while(j<=limit&&F[j].s<=Q[i].a)
		{
			for(register int p=F[j].id,k=1;p<=limit;p+=F[j].id,++k)add(p,F[j].s*mu[k]);
			++j;
		}
		ans[Q[i].id]=solve(Q[i].n,Q[i].m);
	}
	for(register int i=1;i<=T;++i)write(ans[i],'\n');
	return 0;
}