hdu1166（线段树）

来源：互联网发布：北大网络教育好考吗编辑：程序博客网时间：2024/04/30 13:29

题意：给出一列数，之后做四种操作中的一种

(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条命令

思路：先建立线段树，之后相应的进行点修改，query的时候递归分段求就行了

代码：

#include <iostream>#include <cstdio>#include <cstdlib>#include <cstring>#define N 50005using namespace std;int n;int p,q;struct Node{    int sum,l,r;};Node node[N<<2];int num[N];void built(int t,int l,int r){    node[t].l=l;    node[t].r=r;    if(l==r) {        node[t].sum=num[l];        return;    }    int mid=(l+r)>>1;    built(t<<1,l,mid);    built(t<<1|1,mid+1,r);    node[t].sum=node[t<<1].sum+node[t<<1|1].sum;}void add(int t,int l,int r){    int mid=(l+r)>>1;    if(l==r) node[t].sum+=q;    else{        if(p<=mid) add(t<<1,l,mid);        else add(t<<1|1,mid+1,r);        node[t].sum=node[t<<1].sum+node[t<<1|1].sum;    }}int query(int t,int l,int r){    int mid=(node[t].l+node[t].r)>>1,ans=0;    if(node[t].l==l && node[t].r==r) return node[t].sum;    if(r<=mid) ans+=query(t<<1,l,r);    else if(l>mid) ans+=query(t<<1|1,l,r);    else{        ans+=query(t<<1,l,mid);        ans+=query(t<<1|1,mid+1,r);    }    return ans;}int main(){    int t,aaa=1;    char s[10];    scanf("%d",&t);    while(t--)    {        scanf("%d",&n);        memset(node,0,sizeof(node));        for(int i=1;i<=n;i++) scanf("%d",&num[i]);        built(1,1,n);        printf("Case %d:\n",aaa++);        while(scanf("%s",s) && strcmp(s,"End"))        {            scanf("%d %d",&p,&q);            if(s[0]=='A')            {                add(1,1,n);            }            else if(s[0]=='S')            {                q=-q;                add(1,1,n);            }            else            {                int ans=query(1,p,q);                printf("%d\n",ans);            }        }    }    return 0;}

0 0