题意:
有一个长度为5e5的数组a,下标为1到5e5,一开始所有a(i)=0
q次操作,操作有两种:
(1,x,y):a[x]+=y
(2,x,y):询问所有下标模 x 的结果为 y 的位置的值的和
数据范围:
q<=5e5
操作1:1<=x<=5e5,-1000<=y<=1000
操作2:1<=x<=5e5,0<=y<x
解法:
这个数据范围,单个查询要么log要么根号
log好像搞不出来,看了别人的题解发现分块搞
sq=sqrt(5e5)=707
修改的时候:
a[x]+=y,更新出mod<=sq的所有mod的答案,更新过程O(sq)
查询的时候:
当x小于sq的时候,因为预处理过,所以可以O(1)查询
当x大于sq的时候,直接暴力循环,最多循环sq次,O(sq)
总复杂度O(q*sq)
ps:
cin关同步还是会tle,以后遇到大量输入的时候还是要稍微注意一下
code:
#include<bits/stdc++.h>
using namespace std;
const int maxm=5e5+5;
const int sq=(int)sqrt(5e5);
int ans[sq+5][sq+5];
int a[maxm];
signed main(){
int q;scanf("%d",&q);
while(q--){
int d,x,y;scanf("%d%d%d",&d,&x,&y);
if(d==1){
a[x]+=y;
for(int mod=1;mod<=sq;mod++){//将修改a[x]的影响添加到每个(2,mod,x%mod)中,mod<=sq
ans[mod][x%mod]+=y;
}
}else if(d==2){
if(x<=sq){//小于sq的O(1)
printf("%d\n",ans[x][y]);
}else{//大于sq的暴力,最多循环sq次,O(sq)
int sum=0;
for(int i=y;i<=5e5;i+=x){
sum+=a[i];
}
printf("%d\n",sum);
}
}
}
return 0;
}