首页 > 代码库 > BZOJ 3529 SDOI2014 数表 莫比乌斯反演+树状数组
BZOJ 3529 SDOI2014 数表 莫比乌斯反演+树状数组
题目大意:令F(i)为i的约数和,多次询问对于1<=x<=n,1<=y<=m,F(gcd(x,y))<=a的所有数对(x,y),求ΣF(gcd(x,y))%(2^31)
n,m<=10^5,a<=10^9
首先如果不考虑a的限制 令g(i)为1<=x<=n,1<=y<=m,gcd(x,y)=i的数的个数
那么显然有
利用线性筛处理出F(i) 那么答案显然是
治好了我多年的公式恐惧症。。。
现在我们只需要求出的前缀和 这个问题就能在O(√n)的时间内出解
枚举每一个i 枚举i的倍数 暴力即可求出这个函数 然后处理前缀和即可 复杂度是O(nlogn)的
那么现在有了a的限制怎么搞呢?
我们发现对答案有贡献的i只有F(i)<=a的i 那么我们将询问按照a从小到大排序 将F(i)从小到大排序 每次询问将<=a的F(i)按照之前的方式暴力插入 用树状数组来维护这个前缀和
这题就搞出来了
时间复杂度O(nlog^2n+q√nlogn) 有点卡。。。取模那里自然溢出int就行了 最后再对2147483647取一下&即可
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #define M 100100 using namespace std; struct query{ int n,m,a,_id; bool operator < (const query &x) const { return a < x.a ; } }queries[20200]; int T,ans[20200]; pair<int,int>f[M]; int mu[M],prime[M],tot; bool not_prime[M]; void Linear_Shaker() { static int min_factor_a[M],min_factor_sum[M]; /* 12=2*2*3 F(12)=(1+2+4)*(1+3) min_factor_a[12]=2*2=4 min_factor_sum[12]=1+2+4=7 */ int i,j; f[1]=make_pair(1,1);mu[1]=1; for(i=2;i<=100000;i++) { f[i].second=i; if(!not_prime[i]) { prime[++tot]=i; min_factor_a[i]=i; f[i].first=min_factor_sum[i]=i+1; mu[i]=-1; } for(j=1;j<=tot&&prime[j]*i<=100000;j++) { not_prime[prime[j]*i]=1; if(i%prime[j]==0) { f[prime[j]*i].first=f[i].first/min_factor_sum[i]* (min_factor_sum[prime[j]*i]=min_factor_sum[i]+min_factor_a[i]*prime[j]); min_factor_a[prime[j]*i]=min_factor_a[i]*prime[j]; mu[prime[j]*i]=0; break; } f[prime[j]*i].first=f[i].first*(prime[j]+1); min_factor_a[prime[j]*i]=prime[j]; min_factor_sum[prime[j]*i]=prime[j]+1; mu[prime[j]*i]=-mu[i]; } } } namespace BIT{ int c[M]; inline void Update(int x,int y) { for(;x<=100000;x+=x&-x) c[x]+=y; } inline int Get_Ans(int x) { int re=0; for(;x;x-=x&-x) re+=c[x]; return re; } } int Query(int n,int m) { int i,last,re=0; if(n>m) swap(n,m); for(i=1;i<=n;i=last+1) { last=min(n/(n/i),m/(m/i)); re+=(n/i)*(m/i)*(BIT::Get_Ans(last)-BIT::Get_Ans(i-1)); } return re&2147483647; } int main() { int i,j,k; Linear_Shaker(); sort(f+1,f+100000+1); cin>>T; for(i=1;i<=T;i++) scanf("%d%d%d",&queries[i].n,&queries[i].m,&queries[i].a),queries[i]._id=i; sort(queries+1,queries+T+1); for(i=1,j=1;i<=T;i++) { for(;j<=100000&&f[j].first<=queries[i].a;j++) for(k=f[j].second;k<=100000;k+=f[j].second) BIT::Update(k,f[j].first*mu[k/f[j].second]); ans[queries[i]._id]=Query(queries[i].n,queries[i].m); } for(i=1;i<=T;i++) printf("%d\n",ans[i]); return 0; }
BZOJ 3529 SDOI2014 数表 莫比乌斯反演+树状数组
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。