1. 程式人生 > >樹狀陣列模板題 hdu 1166

樹狀陣列模板題 hdu 1166

k為二進位制末尾0的個數

 

樹狀陣列                                                                                二進位制             k        2k

tree[1] = A[1]                                                                          001                  0        1

tree[2] = A[1] + A[2]                                                               010                  1         2

tree[3] = A[3]                                                                          011                  0         1

tree[4] = A[1] +A[2] +A[3]+A[4]                                              100                  2         4

tree[5] = A[5]                                                                          101                  0         1

tree[6] = A[5] +A[6]                                                                 110                  1         2

tree[7] = A[7]                                                                           111                  0         1

tree[8] = A[1] +A[2] +A[3]+A[4]+A[5] +A[6] +A[7] +A[8]          1000                3         8

 

 求和:                                                                       二進位制

sum(1) = tree[1]                                               001             001     

sum(2) = tree[2]                                               010             010

sum(3) = tree[3] + tree[2]                                 011             011 + 010

sum(4) = tree[4]                                               100             100

sum(5) =  tree[5] + tree[4]                                101            101 + 100

sum(6) = tree[6] + tree[4]                                 110             110 + 100

sum(7) = tree[7]+tree[6]+tree[4]                       111             111 + 110 + 100

sum(8) = tree[8]                                               1000            1000

 

 

lowbit操作:

例如7

lowbit(7) = 1;      cur + tree[7]

7 - 1  = 6;

lowbit (6) = 2;     cur+ tree[6]

6 - 2 = 4;

lowbit(4) =  4;     cur+ tree[4]

4 - 4 = 0;

 

程式碼:

#include <bits/stdc++.h>

using namespace std;

const int maxn = 50005;
int tree[maxn];
int n;
int lowbit(int x)
{
    return x&(-x);
}
void add(int i,int value)  //從葉子節點向上更新  例如 add(1,x)  tree[001] -> tree[010] -> tree[100] 需要更新1->2->4
{
//    cout<<"***"<<endl;
    while(i<=n)
    {
        tree[i]+=value;
        i+=lowbit(i);
    }
}
int sum(int i)   //從根節點向下加和  求sum(8) tree[1000] ;;;;; sum(7) tree[111]+tree[110]+tree[100] = tree7+tree6+tree4
{
    int cur = 0;
    while(i>0)
    {
        cur += tree[i];
        i-=lowbit(i);
    }
    return cur;
}
int main()
{
    int T;
    scanf("%d",&T);
    for(int i=1;i<=T;i++)
    {
        memset(tree,0,sizeof(tree));
        scanf("%d",&n);
        for(int j=1;j<=n;j++)
        {
            int x;
            scanf("%d",&x);
            add(j,x);
        }
        printf("Case %d:\n",i);
        string s;
        while(cin>>s)
        {
            if(s=="End")
                break;
            else if(s=="Query")
            {
                int x,y;
                scanf("%d%d",&x,&y);
                printf("%d\n",sum(y)-sum(x-1));
            }
            else if(s=="Add")
            {
                int pos,value;
                scanf("%d%d",&pos,&value);
                add(pos,value);
            }
            else if(s=="Sub"){
                int pos,value;
                scanf("%d%d",&pos,&value);
                add(pos,-value);
            }
        }
    }
    return 0;
}