hdu1166

链接：点击打开链接

题意：第一行一个整数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表示结束,这条命令在每组数据最后出现

对第i组数据,首先输出“Case i:”和回车,

对于每个Query询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内

代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string.h>
using namespace std;
int node[150000];
void update(int pos,int l,int r,int id,int x){
    int mid;
    if(l==r){                            //建数和修改写到一起减少代码长度
        node[pos]+=x;
        return;
    }
    else{
        mid=(l+r)/2;
        if(id<=mid)
        update(2*pos,l,mid,id,x);
        else
        update(2*pos+1,mid+1,r,id,x);
    }
    node[pos]=node[2*pos]+node[2*pos+1];  //回溯时加上左区间和右区间的值
}
int question(int pos,int l,int r,int ll,int rr){
    int mid,sum=0;
    if(ll<=l&&rr>=r)return node[pos];
    else{
        mid=(l+r)/2;
        if(ll<=mid)sum+=question(2*pos,l,mid,ll,rr);
        if(rr>mid)sum+=question(2*pos+1,mid+1,r,ll,rr);
        return sum;
    }
}                                   //以上为线段树模板，因人而异，也可以写成结构体那种形式
int main(){
    int i,k,t,n,x,y;
    int a[50005];
    char str[100];
    scanf("%d",&t);
    for(k=1;k<=t;k++){
        memset(node,0,sizeof(node));
        scanf("%d",&n);
        for(i=1;i<=n;i++){
            scanf("%d",&a[i]);
            update(1,1,n,i,a[i]);
        }
        for(i=1;i<=100;i++)
        cout<<node[i]<<" ";
        cout<<endl;
        printf("Case %d:\n",k);
        while(1){
            cin>>str;
            if(strcmp(str,"End")==0)
            break;
            if(strcmp(str,"Add")==0){
            scanf("%d%d",&x,&y);
            update(1,1,n,x,y);
            }
            if(strcmp(str,"Sub")==0){
            scanf("%d%d",&x,&y);
            update(1,1,n,x,-y);
            }
            if(strcmp(str,"Query")==0){
            scanf("%d%d",&x,&y);
            printf("%d\n",question(1,1,n,x,y));
            }
        }
    }
    return 0;
}

版权声明：本文为博主原创文章，未经博主允许不得转载。