LightOJ1112——Curious Robin Hood

来源：互联网发布：数控加工编程实例编辑：程序博客网时间：2024/05/22 15:59
线段树区间总和，单点更新。
#include <iostream>#include <cstdio>#include <cstdlib>#include <cstring>using namespace std;const int maxN=100010;int tree[4*maxN];int money[maxN];void build(int no,int l,int r){    if(l==r)    {        tree[no]=money[l];        return;    }    int mid=(l+r)>>1;    int no2=no<<1;    build(no2,l,mid);    build(no2+1,mid+1,r);    tree[no]=tree[no2]+tree[no2+1];}void update1(int no,int l,int r,int x){    if(l==r)    {        printf("%d\n",tree[no]);        tree[no]=0;        return;    }    int mid=(l+r)>>1;    int no2=no<<1;    if(x<=mid) update1(no2,l,mid,x);    else update1(no2+1,mid+1,r,x);    tree[no]=tree[no2]+tree[no2+1];}void update2(int no,int l,int r,int x,int change){    if(l==r)    {        tree[no]+=change;        return;    }    int mid=(l+r)>>1;    int no2=no<<1;    if(x<=mid) update2(no2,l,mid,x,change);    else update2(no2+1,mid+1,r,x,change);    tree[no]=tree[no2]+tree[no2+1];}int query(int no,int l,int r,int a,int b){    if(a==l && b==r)    {        return tree[no];    }    int mid=(l+r)>>1;    int no2=no<<1;    if(b<=mid) return query(no2,l,mid,a,b);    if(a>mid) return query(no2+1,mid+1,r,a,b);    return query(no2,l,mid,a,mid)+query(no2+1,mid+1,r,mid+1,b);}int main(){    int t;    //freopen("stdin.txt","r",stdin);    scanf("%d",&t);    for(int id=1; id<=t; id++)    {        int n,m;        int cas;        int x,y;        printf("Case %d:\n",id);        scanf("%d %d",&n,&m);        for(int i=1; i<=n; i++)            scanf("%d",&money[i]);        build(1,1,n);        for(int i=1; i<=m; i++)        {            scanf("%d",&cas);            switch(cas)            {            case 1:                scanf("%d",&x);                x+=1;                update1(1,1,n,x);                break;            case 2:                scanf("%d %d",&x,&y);                x+=1;                update2(1,1,n,x,y);                break;            case 3:                scanf("%d %d",&x,&y);                x+=1,y+=1;                printf("%d\n",query(1,1,n,x,y));                break;            }        }    }    return 0;}
0 0