线段树的应用之敌兵布阵

首先让我们先来了解一下什么是线段树 

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

使用线段树可以快速的查找某一个节点在若干条线段中出现的次数，时间复杂度为O(logN)。而未优化的空间复杂度为2N，实际应用时一般还要开4N的数组以免越界，因此有时需要离散化让空间压缩。

对于线段树中的每一个非叶子节点[a,b]，它的左儿子表示的区间为[a,(a+b)/2]，右儿子表示的区间为[(a+b)/2+1,b]。因此线段树是平衡二叉树，最后的子节点数目为N，即整个线段区间的长度。

基本结构

    

线段树是建立在线段的基础上，每个结点都代表了一条线段[a,b]。长度为1的线段称为元线段。非元线段都有两个子结点，左结点代表的线段为[a,(a + b) / 2]，右结点代表的线段为[((a + b) / 2）+1,b]。

下图就是两棵长度范围为[1,5][1,10]的线段树。

长度范围为[1,L] 的一棵线段树的深度为log (L) + 1。这个显然，而且存储一棵线段树的空间复杂度为O(L）。

线段树支持最基本的操作为插入和删除一条线段。下面以插入为例，详细叙述，删除类似。

将一条线段[a,b] 插入到代表线段[l,r]的结点p中，如果p不是元线段，那么令mid=（l+r）/2。如果b<mid，那么将线段[a,b] 也插入到p的左儿子结点中，如果a>mid，那么将线段[a,b] 也插入到p的右儿子结点中。

插入（删除）操作的时间复杂度为O（logn）。

用于要大量用到输入输出，最好用scanf来进行数据的输入 

 

#include <iostream>
#include <string>
#include<stdio.h>
using namespace std;
struct sd{
    int left;
    int right;
    int sum;
};
#define N 50003
int camp[N],ans;
sd tree[4*N];
void build(int index,int left,int right)
{
    tree[index].left=left;
    tree[index].right=right;
    if(left==right){
        tree[index].sum=camp[left];
    }
    else{
        int mid=(left+right)/2;
        build(index*2,left,mid);
        build(index*2+1,mid+1,right);
        tree[index].sum=tree[index*2].sum+tree[index*2+1].sum;
    }
}
void quiry(int index,int left,int right)
{
    if(left<=tree[index].left&&right>=tree[index].right){
        ans+=tree[index].sum;
    }
    else{
        int mid=(tree[index].left+tree[index].right)/2;
        if(left>mid){
            quiry(index*2+1,left,right);
        }
        else{
            if(right<=mid){
                quiry(index*2,left,right);
            }
            else{
                quiry(index*2,left,right);
                quiry(index*2+1,left,right);
            }
        }
    }

}
void add(int x,int data,int index)
{
    int mid;
    tree[index].sum+=data;
    if(tree[index].left==x&&tree[index].right==x){
        return;
    }
    mid=(tree[index].left+tree[index].right)/2;
    if(x>mid)add(x,data,index*2+1);
    else add(x,data,index*2);
}
int T,n;
int main()
{
    scanf("%d",&T);
    char a[10];
    for(int i=1;i<=T;i++){
        scanf("%d",&n);
        for(int j=1;j<=n;j++){
            scanf("%d",&camp[j]);
        }
        build(1,1,n);
        cout<<"Case "<<i<<":"<<endl;
        int ii,jjj;
        while(scanf("%s",a)&&a[0]!='E'){
               scanf("%d%d",&ii,&jjj);
            if(a[0]=='Q'){
            ans=0;
            quiry(1,ii,jjj);
            cout<<ans<<endl;
        }
        else if(a[0]=='A'){
            add(ii,jjj,1);
        }
        else if(a[0]=='S'){
            add(ii,-jjj,1);
        }
    }

    }
    return 0;
}