有一张N×m的数表,其第i行第j列(1 < =i < =礼,1 < =j < =m)的数值为能同时整除i和j的所有自然数之和。给定a,计算数表中不大于a的数之和。
标签:sdoi upd 数值 getc har output inline size 莫比乌斯反演
有一张N×m的数表,其第i行第j列(1 < =i < =礼,1 < =j < =m)的数值为能同时整除i和j的所有自然数之和。给定a,计算数表中不大于a的数之和。
输入包含多组数据。
输入的第一行一个整数Q表示测试点内的数据组数,接下来Q行,每行三个整数n,m,a(|a| < =10^9)描述一组数据。
对每组数据,输出一行一个整数,表示答案模2^31的值。
1 < =N.m < =10^5 , 1 < =Q < =2×10^4
题解:首先我们要知道一个数的约数和是nloglogn级别的,所以先不考虑a的限制,我们还是采用熟悉的莫比乌斯反演。
$ans=\sum\limits_{i=1}^n\sum\limits_{j=1}^nf(gcd(i,j))\\=\sum\limits_{d=1}^nf(d)\sum\limits_{i=1}^{\lfloor \frac n d \rfloor}\sum\limits_{j=1}^{\lfloor \frac m d \rfloor } [gcd(i,j)==1]\\=\sum\limits_{d=1}^nf(d)\sum\limits_{e=1}^{\lfloor \frac n d \rfloor}\mu(e) \lfloor \frac n {de} \rfloor \lfloor \frac m {de} \rfloor\\=\sum\limits_{D=1}^{n}\sum\limits_{d|D}f(d)\mu(\frac D d) \lfloor \frac n {D} \rfloor \lfloor \frac m {D} \rfloor$
那么如果考虑a的限制呢?我们将所有询问离线,按a排序,然后从小到大处理所有询问,每处理到一个询问,就将所有f(d)<=a的d的贡献都统计出来。也就是说我们需要用某个数据结构来维护f*mu的前缀和,树状数组即可。
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; const int N=100000; typedef unsigned int ui; int pri[N/10]; ui mu[N+10],f[N+10],s[N+10],ans[N]; int q[N+10]; int num,Q; bool np[N+10]; struct node { int n,m,a,org; }p[N]; inline int rd() { int ret=0,f=1; char gc=getchar(); while(gc<‘0‘||gc>‘9‘) {if(gc==‘-‘) f=-f; gc=getchar();} while(gc>=‘0‘&&gc<=‘9‘) ret=ret*10+(gc^‘0‘),gc=getchar(); return ret*f; } bool cmp(const node &a,const node &b) { return a.a<b.a; } bool cmpq(const int &a,const int &b) { return f[a]<f[b]; } inline void updata(int x,ui val) { for(int i=x;i<=N;i+=i&-i) s[i]+=val; } inline ui query(int x) { ui ret=0; for(int i=x;i;i-=i&-i) ret+=s[i]; return ret; } int main() { int i,j,k,last; mu[1]=1; for(i=2;i<=N;i++) { if(!np[i]) pri[++num]=i,mu[i]=-1; for(j=1;j<=num&&i*pri[j]<=N;j++) { np[i*pri[j]]=1; if(i%pri[j]==0) break; mu[i*pri[j]]=-mu[i]; } } for(i=1;i<=N;i++) for(j=i;j<=N;j+=i) f[j]+=i; for(i=1;i<=N;i++) q[i]=i; Q=rd(); for(i=1;i<=Q;i++) { p[i].n=rd(),p[i].m=rd(),p[i].org=i; if(p[i].n>p[i].m) swap(p[i].n,p[i].m); p[i].a=max(rd(),0); } sort(q+1,q+N+1,cmpq); sort(p+1,p+Q+1,cmp); for(i=j=1;i<=Q;i++) { for(;f[q[j]]<=p[i].a;j++) for(k=q[j];k<=N;k+=q[j]) updata(k,f[q[j]]*mu[k/q[j]]); for(k=1;k<=p[i].n;k=last+1) { last=min(p[i].n/(p[i].n/k),p[i].m/(p[i].m/k)); ans[p[i].org]+=(p[i].n/k)*(p[i].m/k)*(query(last)-query(k-1)); } } for(i=1;i<=Q;i++) printf("%u\n",ans[i]&0x7fffffff); return 0; }
【BZOJ3529】[Sdoi2014]数表 莫比乌斯反演+树状数组
标签:sdoi upd 数值 getc har output inline size 莫比乌斯反演
原文地址:http://www.cnblogs.com/CQzhangyu/p/7898351.html