来源:http://acm.hdu.edu.cn/showproblem.php?pid=4267
题意:给一些数,有两种操作,一种是在[a,b] 区间内,对(i - a)% k == 0 的加value,另一种操作是询问某个位置的值。
思路:很容易想到这是一个插线问点的问题,但是在更新值 的时候和平常的更新不同,这道题又增加了一个限制条件。因为k比较小(<=10),余数的情况也很少,因此我们可以开55棵树状数组,用cnt[x][k][mod] 表示x 对k 取余为mod,这样在询问的时候只要循环到10就可以 了。而且(i - a) % k == 0即是 i % k == a % k == mod,这样更新的时候就可以更新(1,b) 和(1,a)了。
代码:
#include <iostream>
#include <cstdio>
#include <string.h>
#include <algorithm>
using namespace std;
const int N = 50010;
int cnt[N][11][11];
int num[N];
int inline lowbit(int x){
return x & (-x);
}
void inline update(int x,int k,int mod,int add){
while(x > 0){
cnt[x][k][mod] += add;
x -= lowbit(x);
}
}
int inline sum(int x,int a){
int s = 0;
while(x < N){
for(int i = 1; i <= 10; ++i){
s += cnt[x][i][a%i];
}
x += lowbit(x);
}
return s;
}
int main(){
//freopen("1.txt","r",stdin);
int n;
while(scanf("%d",&n) != EOF){
for(int i = 1; i <= n; ++i)
scanf("%d",&num[i]);
memset(cnt,0,sizeof(cnt));
int m;
scanf("%d",&m);
while(m--){
int op,a,b,k,add;
scanf("%d",&op);
if(op == 1){
scanf("%d%d%d%d",&a,&b,&k,&add);
update(b,k,a%k,add);
update(a-1,k,a%k,-add);
}
else{
scanf("%d",&a);
int ans = sum(a,a);
printf("%d\n",ans + num[a]);
}
}
}
return 0;
}