线段树(hdu1166)

#include <stdio.h>
#include <math.h>
#include <string.h>
typedef struct STU{
int grade;
int left;
int right;
}STU;
STU st[2097152];
int num[2000001];
int max(int a,int b)
{
    return a>b?a:b;
}
void BuildTree(int i,int left,int right)
 {
     st[i].left=left;
     st[i].right=right;
     st[i].grade=0;//初始化为0
     if (left==right)//如果是一个点,直接打上记号
     {
         num[left]=i;
         return;
     }
     BuildTree(i<<1,left,(int)floor((left+right)/2.0));
     BuildTree((i<<1)+1,(int)floor((left+right)/2.0)+1,right);//注意括号和优先级
 }
void UpdataTree(int r,int b)
{
   if (r==1)
   {
       st[r].grade+=b;
       return;
   }
   st[r].grade=st[r].grade+b;
   UpdataTree(r/2,b);
}
int sum;
void Query(int i,int l,int r)
{
    if ((st[i].left==l)&&(st[i].right==r))
    {
        sum=sum+st[i].grade;
        return;
    }
    i=i<<1;
    if (l<=st[i].right)//如果左边有涉及
    {
        if (r<=st[i].right)//如果全在左部区间
            Query(i,l,r);
        else//不全在左区间
            Query(i,l,st[i].right);//继续查询在左区间的那部分
    }
    i++;
    if (r>=st[i].left)//如果右边有涉及
      {
          if (l>=st[i].left)
            Query(i,l,r);//全在右区间
          else
            Query(i,st[i].left,r);
      }
}
int main()
{
    int T,p;
    int i,j,n,m,c,d;
    char s[10];
    scanf("%d",&T);
    p=T;
    while(T--)
    {
       printf("Case %d:\n",p-T);
       scanf("%d",&n);
       BuildTree(1,1,n);
       for (i=1;i<=n;i++)
       {
           scanf("%d",&m);
           UpdataTree(num[i],m);
       }
      while(scanf("%s",s)&&(strcmp(s,"End")!=0))
      {
        scanf("%d%d",&c,&d);
       if (strcmp(s,"Query")==0)
       {
           sum=0;
           Query(1,c,d);
           printf("%d\n",sum);
       }
       else
        if (strcmp(s,"Add")==0)
       {
           UpdataTree(num[c],d);
       }
       else
        UpdataTree(num[c],-1*d);
      }
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值