线段数模板题

第一行一个整数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条命令 

#include
#include
struct node
{
    int l,r,sum;
}tree[50000*4];
int num[500005],sum;
void pushup(int rt)
{
    tree[rt].sum=tree[2*rt].sum+tree[2*rt+1].sum;
}
void build(int l,int r,int root)//建树
{
    tree[root].l=l;
    tree[root].r=r;
    if(l==r)
    {
        tree[root].sum=num[l];//找到最下面的点,则停止。
        return ;
    }
    int mid=(l+r)/2;//若不是最下面的点的话
    build(l,mid,2*root);
    build(mid+1,r,2*root+1);
    pushup(root);
}
void update(int root,int x,int c)//把第x个数更新成c
{
    if(tree[root].l==x&&tree[root].r==x)
    {
        tree[root].sum+=c;//一旦找到这个数就停止
        return ;
    }
    int mid=(tree[root].l+tree[root].r)/2;
    //tree[].r这种形式就是第几个数,所以现在在找中间的数
    if(x<=mid) update(2*root,x,c);//向左
    else update(2*root+1,x,c);//向右
    pushup(root);
}
void sub(int root,int x,int c)
{
    if(tree[root].l==x&&tree[root].r==x)
    {
        tree[root].sum-=c;
        return ;
    }
    int mid=(tree[root].l+tree[root].r)/2;
    if(x<=mid) sub(2*root,x,c);
    else sub(2*root+1,x,c);
    pushup(root);
}
void query(int l,int r,int root)
{
    if(tree[root].l==l&&tree[root].r==r)
    {
        sum=sum+tree[root].sum;
        return ;
    }
    root*=2;//左树
    if(l<=tree[root].r)
    {
        if(r<=tree[root].r) query(l,r,root);
        else query(l,tree[root].r,root);
    }
    root++;//右树
    if(r>=tree[root].l)
    {
        if(l>=tree[root].l) query(l,r,root);
        else query(tree[root].l,r,root);
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    int cas=0;
    while(t--)
    {
        cas++;
        int n,x,y;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&num[i]);
        }
        char ch[15];
        build(1,n,1);
        printf("Case %d:\n",cas);
        while(~scanf("%s",ch))
        {
            if(strcmp(ch,"End")==0) break;
            if(strcmp(ch,"Query")==0)
            {
                sum=0;
                scanf("%d%d",&x,&y);
                query(x,y,1);
                printf("%d\n",sum);
            }
            if(strcmp(ch,"Add")==0)
            {
                scanf("%d%d",&x,&y);
                update(1,x,y);
            }
            if(strcmp(ch,"Sub")==0)
            {
                scanf("%d%d",&x,&y);
                sub(1,x,y);
            }
        }

    }
}

 

 

 

你可能感兴趣的:(线段数模板题)