【三维树状数组与离散化】HDU 4267——A Simple Problem with Integers

题目：点击打开链接

长春赛区网络赛的题，也是2012年五大区网络赛第一场的第一个题，当时智商拙计，连树状数组和线段树的大名也没有听说过，居然以为是模拟被大神们好一阵修理。。

现在学习了树状数组，写起来也是磕磕绊绊，参阅了若干大神的若干资料，终于凑合了这个三维树状数组的更新。和一般的树状数组不同的是，为节约时间，尽量不TLE，我们的第二维是K，第三维是a%k(为什么是这个呢？原式中的(i-a)%k==0可以移向的。。更新的时候也是按照这个余数来跳着更新，忽忽悠悠就这么过了。。。。

//(i-a)%k==0可以换成i%k==a%k,这个思路转自大神blog 

#include <iostream>
#include <stdio.h>
#include <string.h>

using namespace std;

int numpack[60000],N;
int c[60000][11][11];  //第二维为k,第三维为mod 



int lowbit(int x)
{
	return x&(-x);
}

void update_addnum(int pos,int delta,int mod,int k)  //修改成三维的 
{
	for(int i=pos;i>0;i-=lowbit(i))
	{
		c[i][k][mod]+=delta;
	}
}

void update_subnum(int pos,int delta,int mod,int k)  //修改成三维的 
{
	for(int i=pos;i>0;i-=lowbit(i))
	{
		c[i][k][mod]-=delta;
	}
}


int find_add(int x)
{
	int a=x;
	int result=0;
	for(int i=x;i<=N;i+=lowbit(i))
	{
		for(int j=1;j<=10;j++)
		{
			result+=c[i][j][a%j];      //把第二维遍历一遍 
		}
	}
	return result;
}

int main()
{
	
	while(scanf("%d",&N)!=EOF)
	{
		memset(numpack,0,sizeof(numpack));
		memset(c,0,sizeof(c));
		for(int i=1;i<=N;i++)
		{
			scanf("%d",&numpack[i]);
		}
		
		int opernum,oper,a,b,c,k;
		
		scanf("%d",&opernum);
		while(opernum--)
		{
			scanf("%d",&oper);
			if(oper==1)
			{
				scanf("%d%d%d%d",&a,&b,&k,&c);
				update_addnum(b,c,a%k,k);
				update_subnum(a-1,c,a%k,k);
				
			}
			
			if(oper==2)
			{
				int idx;
				scanf("%d",&idx);
				int res=numpack[idx]+find_add(idx);//更新值加原始值 
				printf("%d\n",res);
			}
			
			
		}
		
		
		
	}
	
	return 0;
}