[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)描述一组数据。
1 < =N.m < =10^5 , 1 < =Q < =2×10^4

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

Sample Input
2
4 4 3
10 10 5

Sample Output
20
148


首先我们不考虑\(a\)的限制,那么题目要求

\[\sum\limits_{i=1}^n\sum\limits_{j=1}^m\sigma_1(\gcd(i,j)) \]

我们直接枚举约数有

\[\sum\limits_{d=1}^n\sum\limits_{i=1}^{\lfloor\frac{n}{d}\rfloor}\sum\limits_{j=1}^{\lfloor\frac{m}{d}\rfloor}\sigma_1(d)[\gcd(i,j)=1] \]

然后我们把\(\sigma_1\)挪到前面,对最后那个式子反演一下有

\[\sum\limits_{d=1}^n\sigma_1(d)\sum\limits_{i=1}^{\lfloor\frac{n}{d}\rfloor}\sum\limits_{j=1}^{\lfloor\frac{m}{d}\rfloor}\sum\limits_{x|i,x|j}\mu(x) \]

\(x\)挪到前面

\[\sum\limits_{d=1}^n\sigma_1(d)\sum\limits_{x=1}^{\lfloor\frac{n}{d}\rfloor}\mu(x)\lfloor\dfrac{n}{dx}\rfloor\lfloor\dfrac{m}{dx}\rfloor \]

我们令\(T=dx\),然后更改枚举顺序有

\[\sum\limits_{T=1}^n\lfloor\dfrac{n}{T}\rfloor\lfloor\dfrac{m}{T}\rfloor\sum\limits_{d|T}\sigma_1(d)\mu(\dfrac{T}{d}) \]

没有\(a\)的限制这题就做完了……但是现实非常骨感

我们设\(g(T)=\sum\limits_{d|T}\sigma_1(d)\mu(\dfrac{T}{d})\),可以发现当\(\sigma_1(d)\leqslant a\)时,才会对\(g(T)\)产生贡献

于是我们将询问按\(a\)从小到大排序,枚举询问的时候,\(a\)变大会使得一些\(\sigma_1(d)\)\(g(T)\)产生贡献,我们就用枚举倍数的方法来找到所有的\(T\),然后我们需要动态修改\(g(T)\)的值,而且还要支持区间询问,因此我们使用常数较小的树状数组实现

假定所有的\(\sigma_1(d)\)都能产生贡献,枚举所有倍数的复杂度为\(\sum\limits_{i=1}^n\dfrac{n}{i}\thickapprox n\ln n\),每次更新\(g(T)\)复杂度为\(\log n\),则修改复杂度为\(O(n\log^2n)\),每次询问需要数论分块,查询区间和复杂度为\(O(\log n)\),所以总复杂度为\(O(q\sqrt n\log n+n\log^2n)\)

/*program from Wolfycz*/
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define Fi first
#define Se second
#define MK make_pair
#define inf 0x7f7f7f7f
#define lowbit(x) ((x)&-(x))
using namespace std;
typedef long long ll;
typedef unsigned int ui;
typedef pair<int,int> pii;
typedef unsigned long long ull;
inline char gc(){
	static char buf[1000000],*p1=buf,*p2=buf;
	return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++;
}
inline int frd(){
	int x=0,f=1; char ch=gc();
	for (;ch<'0'||ch>'9';ch=gc())	if (ch=='-')	f=-1;
	for (;ch>='0'&&ch<='9';ch=gc())	x=(x<<3)+(x<<1)+ch-'0';
	return x*f;
}
inline int read(){
	int x=0,f=1; char ch=getchar();
	for (;ch<'0'||ch>'9';ch=getchar())	if (ch=='-')	f=-1;
	for (;ch>='0'&&ch<='9';ch=getchar())	x=(x<<3)+(x<<1)+ch-'0';
	return x*f;
}
inline void print(int x){
	if (x<0)	putchar('-'),x=-x;
	if (x>9)	print(x/10);
	putchar(x%10+'0');
}
const int N=1e5;
int mu[N+10],prime[N+10],g[N+10];//$f(d)=\sum\limits_{x|d}x$
bool inprime[N+10];
pii f[N+10];
void prepare(){
	mu[1]=1; int tot=0;
	f[1]=MK(1,1);
	for (int i=2;i<=N;i++){
		if (!inprime[i])	prime[++tot]=i,mu[i]=-1,g[i]=i+1,f[i]=MK(i+1,i);
		for (int j=1;j<=tot&&i*prime[j]<=N;j++){
			inprime[i*prime[j]]=1;
			if (i%prime[j]==0){
				mu[i*prime[j]]=0;
				g[i*prime[j]]=g[i]*prime[j]+1;
				f[i*prime[j]]=MK(f[i].Fi/g[i]*g[i*prime[j]],i*prime[j]);
				break;
			}
			mu[i*prime[j]]=-mu[i];
			f[i*prime[j]]=MK(f[i].Fi*f[prime[j]].Fi,i*prime[j]);
			g[i*prime[j]]=prime[j]+1;
		}
	}
}
int tree[N+10];
void Modify(int x,int v){for (;x<=N;x+=lowbit(x))	tree[x]+=v;}
int Query(int x){
	int res=0;
	for (;x;x-=lowbit(x))	res+=tree[x];
	return res;
}
const int M=2e4;
struct S1{
	int n,m,a,ID;
	void rd(int i){n=read(),m=read(),a=read(),ID=i;}
	bool operator <(const S1 &tis)const{return a<tis.a;}
}A[M+10];
int solve(int n,int m){
	if (n>m)	swap(n,m);
	int res=0;
	for (int i=1,pos;i<=n;i=pos+1){
		pos=min(n/(n/i),m/(m/i));
		res=(res+(Query(pos)-Query(i-1))*(n/i)*(m/i));
	}
	return res;
}
int Ans[M+10];
int main(){
	prepare();
	sort(f+1,f+1+N);
	int Q=read();
	for (int i=1;i<=Q;i++)	A[i].rd(i);
	sort(A+1,A+1+Q);
	for (int i=1,j=1;i<=Q;i++){
		while (f[j].Fi<=A[i].a&&j<=N){
			for (int k=f[j].Se;k<=N;k+=f[j].Se)
				Modify(k,f[j].Fi*mu[k/f[j].Se]);
			j++;
		}
		Ans[A[i].ID]=solve(A[i].n,A[i].m);
	}
	for (int i=1;i<=Q;i++)	printf("%d\n",Ans[i]&(~(1<<31)));
	return 0;
}
posted @ 2019-01-16 10:31  Wolfycz  阅读(319)  评论(0编辑  收藏  举报