HDU_1166_敌兵布阵(线段树OR树状数组)

题型：数据结构

题意：中文不多说。

分析：

两种方法：线段树和树状数组。

两者之间的比较：

       线段树是一种二叉搜索树，与区间树相似，它将一个区间划分成一些单元区间，每个单元区间对应线段树中的一个叶结点。

       树状数组是一个可以很高效的进行区间统计的数据结构。在思想上类似于线段树，比线段树节省空间，编程复杂度比线段树低，但适用范围比线段树小。

1、线段树方法：

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
using namespace std;
const int maxn = 1234567;
int a[maxn<<2],value;

void update(int pos,int value,int l,int r,int root) {
    if(l==r) {
        a[root]+=value;
        return;
    } else {
        int mid=(l+r)>>1;
        if(pos<=mid) {
            update(pos,value,l,mid,root<<1);
        } else {
            update(pos,value,mid+1,r,root<<1|1);
        }
        a[root]=a[root<<1]+a[root<<1|1];
        return;
    }
}

int query(int from,int to,int l,int r,int root) {
    if(from<=l&&r<=to) {
        return a[root];
    }
    int mid=(l+r)>>1;
    int tem=0;
    if(from<=mid) {
        tem+=query(from,to,l,mid,root<<1);
    }
    if(mid<to) {
        tem+=query(from,to,mid+1,r,root<<1|1);
    }
    return tem;

}
int main() {
    int t,n,sum;
    int xx,yy;
    char cmd[10];
    scanf("%d",&t);
    for(int k=1; k<t+1; k++) {
        scanf("%d",&n);
        memset(a,0,sizeof(a));
        for(int i=1; i<=n; i++) {
            scanf("%d",&value);
            update(i,value,1,n,1);
        }
        printf("Case %d:\n",k);
        while(~scanf("%s",cmd)&&(cmd[0]!='E')) {
            scanf("%d%d",&xx,&yy);
            if(cmd[0]=='A')
                update(xx,yy,1,n,1);
            if(cmd[0]=='S')
                update(xx,-yy,1,n,1);
            if(cmd[0]=='Q') {
                int temp;
                if(yy<xx) {
                    temp=xx;
                    xx=yy;
                    yy=temp;
                }
                sum=query(xx,yy,1,n,1);
                printf("%d\n",sum);
            }
        }
    }
    return 0;
}

2、树状数组方法：

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
using namespace std;
const int MAXN = 123456;
int n,c[MAXN];
char cmd[10];

//计算2^k,k是x的二进制末尾0的个数；
//设节点编号为x，那么这个节点管辖的区间为2^k个元素
int lowbit(int x){
    return x&(-x);
}

void update(int x,int value){
    while(x<=n){
        c[x]+=value;
        x+=lowbit(x);
    }
}

int sum(int x){
    int s=0;
    while(x>0){
        s+=c[x];
        x-=lowbit(x);
    }
    return s;
}

int main(){
    int t,time=1;
    scanf("%d",&t);
    int value;
    while(t--){
        printf("Case %d:\n",time);;
        time++;
        memset(c,0,sizeof(c));
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            scanf("%d",&value);
            update(i,value);
        }
        int a,b;
        while(1){
            scanf("%s",cmd);
            if(cmd[0]=='E') break;
            else if(cmd[0]=='Q'){
                scanf("%d%d",&a,&b);
                printf("%d\n",sum(b)-sum(a-1));
            }
            else if(cmd[0]=='A'){
                scanf("%d%d",&a,&b);
                update(a,b);
            }
            else{
                scanf("%d%d",&a,&b);
                update(a,-b);
            }
        }
    }
}