【刷题】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} floor lfloor frac{m}{d} floor)
继续推

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

(T=ik)

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

前面部分整除分块,我们只要处理出后半部分的前缀和就行了
对于(F(k))直接(O(nlog_2n))暴力枚举约数和它的倍数
对于(mu)这就不用说了吧。。。(看之前的文章)
好的,我们把改装之后的问题解决了,可是原问题呢?
回到真正的(ans),因为有a的限制,所以那些(F(i))大于a的是不能加贡献的
那么我们把询问按a排序,然后用树状数组维护(F(k)mu(lfloor frac{T}{k} floor))的前缀和,只有当前询问的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='')
{
	if(x<0)putchar('-'),x=-x;
	if(x>9)write(x/10);
	putchar(x%10+'0');
	if(c!='')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],'
');
	return 0;
}

原文地址:https://www.cnblogs.com/hongyj/p/8551006.html