hdu1166

来源：互联网发布：建筑学知乎编辑：程序博客网时间：2024/05/17 01:23

链接：点击打开链接

题意：第一行一个整数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以内

线段树单点更新，求和

代码1：

#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;}

代码2：

#include <stdio.h>#include <string.h>#include <iostream>#include <algorithm>using namespace std;const int SIZE=50005;int segtree[SIZE<<2];void build(int l,int r,int rt){    int m;    if(l==r){        scanf("%d",&segtree[rt]);        return;    }    m=(l+r)>>1;    build(l,m,rt<<1);    build(m+1,r,rt<<1|1);    segtree[rt]=segtree[rt<<1]+segtree[rt<<1|1];}void update(int p,int add,int l,int r,int rt){    int m;    if(l==r){        segtree[rt]+=add;        return;    }    m=(l+r)>>1;    if(p<=m)    update(p,add,l,m,rt<<1);    else    update(p,add,m+1,r,rt<<1|1);    segtree[rt]=segtree[rt<<1]+segtree[rt<<1|1];}int query(int L,int R,int l,int r,int rt){    int m,ans;    if(L<=l&&r<=R)    return segtree[rt];    m=(l+r)>>1;    ans=0;    if(L<=m)    ans+=query(L,R,l,m,rt<<1);    if(R>m)    ans+=query(L,R,m+1,r,rt<<1|1);    return ans;}int main(){    int t,n,i,j,k,a,b;    char str[100];    scanf("%d",&t);    for(k=1;k<=t;k++){        scanf("%d",&n);        build(1,n,1);        printf("Case %d:\n",k);        while(scanf("%s",str)&&strcmp(str,"End")!=0){            scanf("%d%d",&a,&b);            if(strcmp(str,"Query")==0)            printf("%d\n",query(a,b,1,n,1));            else if(strcmp(str,"Add")==0)            update(a,b,1,n,1);            else            update(a,-b,1,n,1);        }    }    return 0;}//强烈推荐看这个讲的线段树，代码特别清晰，适合作为模板//http://blog.csdn.net/metalseed/article/details/8039326

0 0