HDU 4267 A Simple Problem with Integers --树状数组

题意：给一个序列，操作1：给区间[a,b]中(i-a)%k==0的位置 i 的值都加上val  操作2：查询 i 位置的值

解法：树状数组记录更新值。 由 (i-a)%k == 0 得知 i%k == a%k，又因为k <= 10，想到建55棵树状数组，即对每个(k,x%k)都建一棵树状数组，每次更新时，在第(k,a%k)棵树状数组上更新a这个点，更新值为val，然后再b+1处更新值为-val，即在[a,b]内更新了val。

查询pos的时候，求出每一个树状数组(k,pos%k)的sum值即可。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
using namespace std;
#define N 50007

int c[N][11][11];
int n,ka[N];

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

void modify(int pos,int k,int mod,int val)
{
    if(pos == 0) return;
    while(pos <= n)
    {
        c[pos][k][mod] += val;
        pos += lowbit(pos);
    }
}

int getsum(int pos)
{
    int res = 0;
    int tmp = pos;
    while(pos > 0)
    {
        for(int i=1;i<=10;i++)
            res += c[pos][i][tmp%i];
        pos -= lowbit(pos);
    }
    return res;
}

int main()
{
    int m,a,b,k,val,i,j;
    int op;
    while(scanf("%d",&n)!=EOF)
    {
        for(i=1;i<=n;i++)
            scanf("%d",&ka[i]);
        memset(c,0,sizeof(c));
        scanf("%d",&m);
        while(m--)
        {
            scanf("%d",&op);
            if(op == 1)
            {
                scanf("%d%d%d%d",&a,&b,&k,&val);
                modify(a,k,a%k,val);
                modify(b+1,k,a%k,-val);
            }
            else
            {
                scanf("%d",&a);
                printf("%d\n",ka[a] + getsum(a));
            }
        }
    }
    return 0;
}

View Code