hdu 4267 多维树状数组

题意：有一个序列

"1 a b k c" means adding c to each of Ai which satisfies a <= i <= b and (i - a) % k == 0. (1 <= a <= b <= N, 1 <= k <= 10, -1,000 <= c <= 1,000)
"2 a" means querying the value of Aa. (1 <= a <= N)

分析转自：http://www.cnblogs.com/Griselda/archive/2012/09/15/2686133.html

感觉是树状数组了

但是树状数组是单点更新而题目要求是在a b 内每隔 k 个数更新..

如果用for循环对a b 内每隔 k 个位置的数更新就会很费时间..

所以有一个方法就是按 k 值 i%k 值还有 i/k+1 值把每隔 k 个位置的值都分组..即按 k 值和 i%k 值建树状数组

到时候维护其中一棵树状数组就好了..

维护的方法是在c[ k ][a%k][(i-a)/k+1] + 1 在c[ k ][a % k ][ (a/k) + (b-a)/k + 2 ] - 1..

关于建成的树状数组..

更新时有55种情况
1,2,3,4,5......
1,3,5,7,9......
2,4,6,8,10....
1,4,7,10,13...
2,5,9,12,15...
3,6,10,13,16...
.
.
.
10,20,30,40,50...

所以用55个树状数组.

建出来的树并不是普通的1,2,3,4,这样的，而是有着间隔的，其实也就是建造了许多树状数组

 #include<stdio.h>

 #include<queue>

 #include<iostream>

 #include<algorithm>

 #include<string.h>

 const int MAXN=;

 int c[][][MAXN];

 int n;

 int lowbit(int x)

 {

     return x&(-x);

 }

 void update(int t1,int t2,int i,int val)

 {

     while(i<=n)

     {

         c[t1][t2][i]+=val;

         i+=lowbit(i);

     }

 }

 int sum(int t1,int t2,int i)

 {

     int s=;

     while(i>)

     {

         s+=c[t1][t2][i];

         i-=lowbit(i);

     }

     return s;

 }

 int num[MAXN];

 int main()

 {

     #ifndef ONLINE_JUDGE

     freopen("1.in","r",stdin);

     #endif

     int m;

     while(scanf("%d",&n)!=EOF)

     {

         for(int i=;i<n;i++)scanf("%d",&num[i]);

         for(int i=;i<;i++)

           for(int j=;j<;j++)

             for(int k=;k<MAXN;k++)

               c[i][j][k]=;

         scanf("%d",&m);

         int a,b,k,q;

         int t;

         while(m--)

         {

             scanf("%d",&t);

             if(t==)

             {

                 scanf("%d%d%d%d",&a,&b,&k,&q);

                 a--;

                 b--;

                 int num=(b-a)/k;

                 int s=a%k-;

                 update(k,s,a/k+,q);    //s是该数组的开头数字，a/k+1是结尾的数字

                 update(k,s,a/k+num+,-q);

             }

             else

             {

                 scanf("%d",&a);

                 a--;

                 int ss=num[a];

                 for(int i=;i<=;i++)

                 {

                     ss+=sum(i,a%i-,a/i+);

                 }

                 printf("%d\n",ss);

             }

         }

     }

     return ;

 }

巴特西

hdu 4267 多维树状数组

最新文章

热门文章