HDU 1166 敌兵布阵 线段树

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

 

线段树模版

 

代码：

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

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

intmap[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;
}

voidBuild(int l,int r,Treep)
{
   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);
}

intQuery(int a,int b,Treep)
{
   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);
}

voidAdd(int a,int b,Treep)
{
   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);
   }
      
}

intmain()
{
   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));
         elseif(strcmp(str,"Add")==0)
             Add(a,b,root);
         else
             Add(a,-b,root);
      }
   }
   return 0;
}