线段树模板

听说是最标准的模板:(敌兵布阵实例)

#include<stdio.h>
#define  max 50000
struct Node
{
    int left;
    int right;
    int sum;
}segTree[4*max+10];
int p[max+10];
void CreatsegTree(int index,int lef,int rig) //创建一个[lef,rig]的线段树。
{
    segTree[index].left=lef;
    segTree[index].right=rig;
    if(lef==rig)
    {
        segTree[index].sum=p[lef];
        return;
    }
    int mid=(lef+rig)>>1;
    int nindex=index<<1;
    CreatsegTree(nindex,lef,mid);
    CreatsegTree(nindex+1,mid+1,rig);
    segTree[index].sum=segTree[nindex].sum+segTree[nindex+1].sum;
}
void serTree(int index,int k,int num)  //更新[k,k]的位置更新线段树的值。
{
    if(k==segTree[index].left&&k==segTree[index].right) //找到[k,k]为更新。
    {
        segTree[index].sum+=num;
        return;
    }
    int mid=(segTree[index].left+segTree[index].right)>>1;
    if(k<=mid)
    {
        serTree(index*2,k,num);
        segTree[index].sum+=num; //底层的位置数更新了,上层的一起更新。
    }
    else
    {
        serTree(index*2+1,k,num);
        segTree[index].sum+=num; //底层的位置数更新了,上层的一起更新。
    }
}
int ans=0;
void Query(int index,int a,int b)  //计算[a,b]段的和。
{
    int mid=(segTree[index].left+segTree[index].right)>>1;

    if(segTree[index].left==a&&segTree[index].right==b)
    {
        ans+=segTree[index].sum;
        return;
    }
    if(segTree[index].left==segTree[index].right) return;
    if(b>mid&&a>mid) Query(index*2+1, a, b);
    else if(a<=mid&&b<=mid) Query( index*2, a, b);
    else
    {
        Query( index*2, a, mid);
        Query( index*2+1, mid+1, b);
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    int cas=1;
    while(t--)
    {
        int n,i;
        scanf("%d",&n);
        for(i=1;i<=n;i++) scanf("%d",p+i);
        CreatsegTree(1,1,n);
        char str[10];
        printf("Case %d:\n",cas++);
        while(~scanf("%s",str)&&str[0]!='E')
        {
            int a ,b;
            scanf("%d%d",&a,&b);
            ans=0;
            switch(str[0])
            {
                case 'A': serTree( 1, a,b);break;
                case 'S': serTree( 1, a,-b);break;
                case 'Q':
                    Query(1, a, b);
                    printf("%d\n",ans);
                    break;
            }
        }
    }
    return 0;
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值