【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

img

题目分析

(h)为预处理的约数之和,根据之前【BZOJ4407】于神之怒加强版的化简套路,我们可以得到

[egin{split} ans&=sum_{i=1}^nsum_{j=1}^mh(gcd(i,j))\ &=sum_{T=1}^nlfloorfrac nT floorlfloorfrac mT floorsum_{d|T}mu(frac Td)h(d) end{split} ]

按正常情况,我们应该把后面的(sumlimits_{d|T}mu(frac Td)h(d))预处理出前缀和形式。

然而,由于现在只有(h(d)<a)的部分才会产生贡献,所以我们可以将询问离线下来,按(a)排序,把不断更新的前缀和存在树状数组中即可。

代码实现

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdio>
#include<iomanip>
#include<cstdlib>
#include<queue>
#define MAXN 0x7fffffff
typedef long long LL;
const int N=100005;
const LL mod=1ll<<31; 
using namespace std;
inline int Getint(){register int x=0,f=1;register char ch=getchar();while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();}return x*f;}
int mu[N],prime[N];
int h[N],low[N];
bool vis[N];

int ksm(int x,int k){
	int ret=1;
	while(k){
		if(k&1)ret=ret*x;
		x=x*x,k>>=1;
	}
	return ret;
}

struct Que{int n,m,a,id;bool operator <(const Que &q)const{return a<q.a;}}s[N];

#define mp(x,y) make_pair(x,y)
#define Pr pair<int,int>
priority_queue<Pr,vector<Pr>,greater<Pr> >q;

#define lowbit(x) x&(-x)
int tr[N],ans[N];
void Add(int x,int val){for(int i=x;i<=1e5;i+=lowbit(i))tr[i]=(1ll*tr[i]+val)%mod;}
int Query(int x){LL ret=0;for(int i=x;i;i-=lowbit(i))ret=(ret+tr[i])%mod;return (ret+mod)%mod;}
int main(){
	mu[1]=h[1]=1;
	q.push(mp(1,1));
	for(int i=2;i<=1e5;i++){
		if(!vis[i])prime[++prime[0]]=i,mu[i]=-1,h[i]=1+i,low[i]=2;
		for(int j=1;j<=prime[0]&&1ll*i*prime[j]<=1e5;j++){
			vis[i*prime[j]]=1;
			if(i%prime[j]==0){
				low[i*prime[j]]=low[i]+1;
				h[i*prime[j]]=1ll*h[i]*(ksm(prime[j],low[i]+1)-1)/(ksm(prime[j],low[i])-1);
				break;
			}
			low[i*prime[j]]=2;
			h[i*prime[j]]=h[i]*h[prime[j]];
			mu[i*prime[j]]=-mu[i];
		}
		q.push(mp(h[i],i));
	}
	
	int Q=Getint();
	for(int i=1;i<=Q;i++)s[i].n=Getint(),s[i].m=Getint(),s[i].a=Getint(),s[i].id=i;
	sort(s+1,s+1+Q);
	
	for(int t=1;t<=Q;t++){
		int n=s[t].n,m=s[t].m;
		if(n>m)swap(n,m);
		while(!q.empty()&&q.top().first<=s[t].a){
			int x=q.top().first,i=q.top().second;q.pop();
			for(int j=i;j<=1e5;j+=i)Add(j,mu[j/i]*x);
		}
		for(int l=1,r,i=s[t].id;l<=n;l=r+1){
			r=min(n/(n/l),m/(m/l));
			ans[i]=(ans[i]+1ll*(n/l)*(m/l)%mod*(Query(r)-Query(l-1))%mod)%mod;
		}
	}
	for(int i=1;i<=Q;i++)cout<<(ans[i]+mod)%mod<<'
';
	return 0;
}
原文地址:https://www.cnblogs.com/Emiya-wjk/p/10008632.html