HDU 1166 敌兵布阵 线段树

标签：c++

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define qq  50005
struct node{
   int left,right,value;
}q[4*qq];                              //根据二叉树性质   数组大小不能少于节点数
int w[qq];
void build(int l,int r,int i)          // 建树
{
    q[i].left=l;
    q[i].right=r;
    if(l==r)
    {
        q[i].value=w[l];                        //最小的节点代表线段(x,x);
        return;
    }
    int m=(l+r)/2;
    build(l,m,2*i);                            // 建左子节点和右子节点
    build(m+1,r,2*i+1);
    q[i].value=q[i*2].value+q[i*2+1].value;     //父节点为两个子节点之和
}
void update(int i,int x,int op,int num)        //更新线段树
{
    if(q[i].left==x&&q[i].right==x)
    {
        if(op==1)
            q[i].value+=num;
        else
            q[i].value-=num;
        return;
    }
    int m=(q[i].left+q[i].right)/2;
    if(x<=m)                                 //  如果目标点在该节点的左节点
        update(2*i,x,op,num);
    else
        update(2*i+1,x,op,num);
    if(op==1)                                // 回溯回来父节点也要更新
        q[i].value+=num;
    else
        q[i].value-=num;
}
int query(int l,int r,int i)                 // 询问树种的值
{
    if(q[i].left==l&&q[i].right==r)
    {
        return q[i].value;                    
    }
    int m=(q[i].left+q[i].right)/2;
    if(m>=r)                                 //如果要询问的右边在该节点的左边
        return query(l,r,2*i);
    else if(l>m)
        return query(l,r,2*i+1);
    else
        return query(l,m,2*i)+query(m+1,r,2*i+1);
}
int main()
{
    int t,i,j,n,a,b,s;
    char ww[20];
    cin>>t;
    for(i=1;i<=t;i++)
    {
        printf("Case %d:\n",i);
        scanf("%d",&n);
        for(j=1;j<=n;j++)
            scanf("%d",&w[j]);
            build(1,n,1);
        while(scanf("%s",ww))
        {
            if(ww[0]=='E')
                break;
            scanf("%d%d",&a,&b);
            if(ww[0]=='Q')
            {
                s=query(a,b,1);
                cout<<s<<endl;
            }
            else if(ww[0]=='A')
            {
                update(1,a,1,b);
            }
            else
            {
                update(1,a,2,b);
            }
        }
    }
    return 0;
}

1
10
1 2 3 4 5 6 7 8 9 10
Query 1 3
Add 3 6
Query 2 7
Sub 10 2
Add 6 3
Query 3 10
End 

Case 1:
6
33
59

充分体现了 数据结构 对时间的优化
线段树是二叉树的一种 每个节点都代表一条线段。。
代码：
 

HDU 1166 敌兵布阵 线段树,布布扣,bubuko.com

HDU 1166 敌兵布阵 线段树

标签：c++

原文地址：http://blog.csdn.net/axuan_k/article/details/38395939