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 }