hdu 1166 敌兵布阵（线段树，伸展树）

敌兵布阵

Time Limit:1000MS    Memory Limit:32768KB    64bit IO Format:%I64d & %I64u

Description

C国的死对头A国这段时间正在进行军事演习，所以C国间谍头子Derek和他手下Tidy又开始忙乎了。A国在海岸线沿直线布置了N个工兵营地,Derek和Tidy的任务就是要监视这些工兵营地的活动情况。由于采取了某种先进的监测手段，所以每个工兵营地的人数C国都掌握的一清二楚,每个工兵营地的人数都有可能发生变动，可能增加或减少若干人手,但这些都逃不过C国的监视。
中央情报局要研究敌人究竟演习什么战术,所以Tidy要随时向Derek汇报某一段连续的工兵营地一共有多少人,例如Derek问:“Tidy,马上汇报第3个营地到第10个营地共有多少人!”Tidy就要马上开始计算这一段的总人数并汇报。但敌兵营地的人数经常变动，而Derek每次询问的段都不一样，所以Tidy不得不每次都一个一个营地的去数，很快就精疲力尽了，Derek对Tidy的计算速度越来越不满:"你个死肥仔，算得这么慢，我炒你鱿鱼!”Tidy想：“你自己来算算看，这可真是一项累人的工作!我恨不得你炒我鱿鱼呢!”无奈之下，Tidy只好打电话向计算机专家Windbreaker求救,Windbreaker说：“死肥仔，叫你平时做多点acm题和看多点算法书，现在尝到苦果了吧!”Tidy说："我知错了。。。"但Windbreaker已经挂掉电话了。Tidy很苦恼，这么算他真的会崩溃的，聪明的读者，你能写个程序帮他完成这项工作吗？不过如果你的程序效率不够高的话，Tidy还是会受到Derek的责骂的.

 

Input

第一行一个整数T，表示有T组数据。
每组数据第一行一个正整数N（N<=50000）,表示敌人有N个工兵营地，接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人（1<=ai<=50）。
接下来每行有一条命令，命令有4种形式：
(1) Add i j,i和j为正整数,表示第i个营地增加j个人（j不超过30）
(2)Sub i j ,i和j为正整数,表示第i个营地减少j个人（j不超过30）;
(3)Query i j ,i和j为正整数,i<=j，表示询问第i到第j个营地的总人数;
(4)End 表示结束，这条命令在每组数据最后出现;
每组数据最多有40000条命令

 

Output

对第i组数据,首先输出“Case i:”和回车,
对于每个Query询问，输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。

 

Sample Input

      
      
      
      

       
       
       
       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 
      
      
      
      

 

Sample Output

      
      
      
      

       
       
       
       Case 1:
6
33
59 
      
      
      
      

 

线段树和伸展树两种做法都试试
线段树快30+ms

线段树：

#include<cstring>
#include<string>
#include<iostream>
#include<queue>
#include<cstdio>
#include<algorithm>
#include<map>
#include<cstdlib>
#include<cmath>
#include<vector>
//#pragma comment(linker, "/STACK:1024000000,1024000000");

using namespace std;

#define INF 0x3f3f3f3f
#define maxn 4*50000

struct node
{
    int l,r;
    int sum;
} T[maxn];

void init(int l,int r,int k)
{
    T[k].l=l;
    T[k].r=r;
    if(l==r)
    {
        scanf("%d",&T[k].sum);
        return ;
    }
    int mid=(l+r)>>1;
    init(l,mid,2*k);
    init(mid+1,r,2*k+1);
    T[k].sum=T[2*k].sum+T[2*k+1].sum;
}

void Insert(int p,int d,int k)
{
    if(T[k].l==T[k].r&&T[k].r==p)
    {
        T[k].sum+=d;
        return ;
    }
    int mid=(T[k].l+T[k].r)>>1;
    if(p<=mid) Insert(p,d,2*k);
    else if(p>mid) Insert(p,d,2*k+1);
    T[k].sum=T[2*k].sum+T[2*k+1].sum;
}

int query(int l,int r,int k)
{
    if(T[k].l==l&&T[k].r==r)
    {
        return T[k].sum;
    }
    int mid=(T[k].l+T[k].r)>>1;
    if(r<=mid) return query(l,r,2*k);
    else if(l>mid) return query(l,r,2*k+1);
    else
    {
        return query(l,mid,2*k)+query(mid+1,r,2*k+1);
    }
}

int main()
{
    int c=1;
    int T;
    scanf("%d",&T);
    while(T--)
    {
        printf("Case %d:\n",c++);
        char str[10];
        int n;
        scanf("%d",&n);
        init(1,n,1);
        while(1)
        {
            scanf("%s",str);
            if(!strcmp(str,"Add"))
            {
                int a,b;
                scanf("%d%d",&a,&b);
                Insert(a,b,1);
            }
            else if(!strcmp(str,"Sub"))
            {
                int a,b;
                scanf("%d%d",&a,&b);
                Insert(a,-b,1);
            }
            else if(!strcmp(str,"Query"))
            {
                int a,b;
                scanf("%d%d",&a,&b);
                int ans=query(a,b,1);
                printf("%d\n",ans);
            }
            else break;
        }
    }
    return 0;
}

splay：

#include<cstring>
#include<string>
#include<iostream>
#include<queue>
#include<cstdio>
#include<algorithm>
#include<map>
#include<cstdlib>
#include<cmath>
#include<vector>
//#pragma comment(linker, "/STACK:1024000000,1024000000");

using namespace std;

#define INF 0x3f3f3f3f
#define keytree (ch[ch[root][1]][0])
#define maxn 50005

int ch[maxn][2],sum[maxn],pre[maxn],val[maxn],tmp[maxn],siz[maxn];
int top,root;

inline void newnode(int &x,int p,int v)
{
    x=++top;
    ch[x][0]=ch[x][1]=0;
    sum[x]=v;
    pre[x]=p;
    val[x]=v;
    siz[x]=1;
}

inline void pushup(int x)
{
    sum[x]=sum[ch[x][0]]+sum[ch[x][1]]+val[x];
    siz[x]=siz[ch[x][0]]+siz[ch[x][1]]+1;
}

inline void build(int &x,int l,int r,int p)
{
    int mid=(l+r)>>1;
    newnode(x,p,tmp[mid]);
    if(l<mid) build(ch[x][0],l,mid-1,x);
    if(r>mid) build(ch[x][1],mid+1,r,x);
    pushup(x);
}

inline void init(int n)
{
    for(int i=0; i<n; i++) scanf("%d",&tmp[i]);
    ch[0][0]=ch[0][1]=siz[0]=pre[0]=sum[0]=val[0]=0;
    root=top=0;
    newnode(root,0,-1);
    newnode(ch[root][1],root,-1);
    build(keytree,0,n-1,ch[root][1]);
    pushup(ch[root][1]);
    pushup(root);
}

inline void Rotate(int x,int kind)
{
    int y=pre[x];
    ch[y][!kind]=ch[x][kind];
    pre[ch[x][kind]]=y;
    if(pre[y]) ch[pre[y]][ch[pre[y]][1]==y]=x;
    pre[x]=pre[y];
    ch[x][kind]=y;
    pre[y]=x;
    pushup(y);
}

void splay(int x,int goal)
{
    while(pre[x]!=goal)
    {
        if(pre[pre[x]]==goal)
            Rotate(x,ch[pre[x]][0]==x);
        else
        {
            int y=pre[x];
            int kind=ch[pre[y]][0]==y;
            if(ch[y][kind]==x)
            {
                Rotate(x,!kind);
                Rotate(x,kind);
            }
            else
            {
                Rotate(y,kind);
                Rotate(x,kind);
            }
        }
    }
    pushup(x);
    if(!goal) root=x;
}

inline int get_kth(int x,int k)
{
    int t=siz[ch[x][0]]+1;
    if(t==k) return x;
    if(k<t) return get_kth(ch[x][0],k);
    else return get_kth(ch[x][1],k-t);
}

int main()
{
    int T;
    scanf("%d",&T);
    int c=1;
    while(T--)
    {
        printf("Case %d:\n",c++);
        int n;
        scanf("%d",&n);
        init(n);
        char str[10];
        while(1)
        {
            int a,b;
            scanf("%s",str);
            if(!strcmp(str,"Query"))
            {
                scanf("%d%d",&a,&b);
                splay(get_kth(root,a),0);
                splay(get_kth(root,b+2),root);
                int ans=sum[keytree];
                printf("%d\n",ans);
            }
            else if(!strcmp(str,"Add"))
            {
                scanf("%d%d",&a,&b);
                splay(get_kth(root,a+1),0);
                val[root]+=b;
                pushup(root);
            }
            else if(!strcmp(str,"Sub"))
            {
                scanf("%d%d",&a,&b);
                splay(get_kth(root,a+1),0);
                val[root]-=b;
                pushup(root);
            }
            else break;
        }
    }
    return 0;
}