HDU 1166 敌兵布阵 线段树

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1166

 

线段树模版

 

代码：

#include<stdio.h>
#include<string.h>
#include<stdlib.h>

typedef struct node{
   int left,right,sum;
   struct node *l,*r;
}*Tree,Node;

int map[50008],num[50008];
Tree root;

Tree NEW()
{
   Tree p;
   p=(Tree)malloc(sizeof(Node));
   p->l=NULL;p->r=NULL;
   p->left=0;p->right=0;p->sum=0;
   return p;
}

void Build(int l,int r,Tree p)
{
   if(l==r)return;
   int mid=(l+r)>>1;
   Tree s,e;
   s=NEW();
   p->l=s;
   s->sum=num[mid]-num[l-1];
   s->left=l;
   s->right=mid;
   Build(l,mid,p->l);
   e=NEW();
   p->r=e;
   e->sum=num[r]-num[mid];
   e->left=mid+1;e->right=r;
   Build(mid+1,r,p->r);
}

int Query(int a,int b,Tree p)
{
   if(a==p->left && b==p->right)return p->sum;
   int mid=(p->left+p->right)/2;
   if(b<=mid)
       return Query(a,b,p->l);
   else if(a>=mid+1)
       return Query(a,b,p->r);
   else return Query(a,mid,p->l) + Query(mid+1,b,p->r);
}

void Add(int a,int b,Tree p)
{
   if(p->left==a && p->right==a){p->sum+=b;return;}
   if(p->left<=a && p->right>=a)
   {
       p->sum+=b;
       Add(a,b,p->l);
       Add(a,b,p->r);
   }
       
}

int main()
{
   int ncase,i,n,T;
   int a,b;
   char str[10];
   scanf("%d",&ncase);
   for(T=1;T<=ncase;T++)
   {
       root=NEW();
       scanf("%d",&n);
       num[0]=0;
       for(i=1;i<=n;i++)
       {
           scanf("%d",&map[i]);
           num[i]=num[i-1]+map[i];
       }
       root->left=1;root->right=n;root->sum=num[n];
       Build(1,n,root);
       printf("Case %d:\n",T);
       while(scanf("%s",str),strcmp(str,"End"))
       {
           scanf("%d%d",&a,&b);
           if(strcmp(str,"Query")==0)
               printf("%d\n",Query(a,b,root));
           else if(strcmp(str,"Add")==0)
               Add(a,b,root);
           else
               Add(a,-b,root);
       }
   }
   return 0;
}