HDU 4267 A Simple Problem with Integers 树状数组

来源：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;
}