hdu 4267 A Simple Problem with Integers

题目链接：hdu 4267 A Simple Problem with Integers

类似于题目：hdu 1556 Color the ball 的技巧实现树状数组的段更新点查询。

由于该题对于段的更新并不是连续的，从而可以构造多个树状数组。因为$k \in [1,10] $，从而可以把更新划分为如下类型：

1,2,3,4,5...

-------------

1,3,5,7,9...

2,4,6,8,10...

-------------

1,4,7,10,13...

2,5,8,11,14...

3,6,9,12,15...

-------------

...

一共55棵树状数组。对于每一种更新，相应的更新该类型对应的树状数组。

在查询时，同样寻找符合该查询类型的树状数组并最终加上初始值即可完成。

代码如下：

 1 #include <cstdlib>
 2 #include  <cstdio>
 3 #include  <cstring>
 4 #include  <iostream>
 5 #define     MAXN 50005
 6 using namespace std;
 7 int bit[12][12][MAXN];
 8 int arr[MAXN];
 9 int N;
10 int lowbit(int x)
11 {
12     return x&(-x);
13 }
14 int query(int a, int b, int l)
15 {
16     int sum = 0;
17     while( l > 0 )
18     {
19         sum += bit[a][b][l];
20         l -= lowbit(l);
21     }
22     return sum;
23 }
24 void add(int a, int b, int l, int x)
25 {
26     while( l <= N )
27     {
28         bit[a][b][l] += x;
29         l += lowbit(l);
30     }
31 }
32 int main(int argc, char *argv[])
33 {
34     while(scanf("%d", &N)!= EOF)
35     {
36         memset(bit, 0, sizeof(bit));
37         for( int i = 1 ; i <= N ; i++ )
38         {
39             scanf("%d", &arr[i]);
40         }
41         int Q;
42         scanf("%d", &Q);
43         for( int i = 0 ; i < Q ; i++ )
44         {
45            int ty;
46            scanf("%d", &ty);
47            if( ty == 1 )
48            {
49                int a, b, k, c;
50                scanf("%d%d%d%d", &a, &b, &k, &c);
51                add(k, a%k, a, c);
52                add(k, a%k, b+1, -c);
53            }
54            if( ty == 2 )
55            {
56                int l;
57                scanf("%d", &l);
58                int ans = arr[l];
59                for( int i = 1 ; i <= 10 ; i++ )
60                {
61                    ans += query(i, l%i, l);
62                }
63                printf ( "%d\n", ans );
64            }
65         }
66     }
67     return 0;
68 }