计蒜客习题:斑点蛇

题目:

有一种神奇斑点蛇，蛇如其名，全身都是斑点，斑点数量可以任意改变。
有一天，蒜头君十分的无聊，开始数蛇上的斑点。假设这条蛇的长度是Ncm，蒜头君已经数完开始时蛇身的第icm上有ai个斑点。
现在蒜头君想知道这条斑点蛇的任意区间的蛇身上一共有多少个斑点。这好像是一个很容易的事情，但是这条蛇好像是和蒜头君过不去，总是刻意的改变蛇身上的斑点数量。
于是，蒜头君受不了了，加上蒜头君有密集型恐惧症。聪明又能干的你能帮帮他吗？

输入格式

第一行一个正整数N（N≤50000）表示这条斑点蛇长度为N厘米，接下来有N个正整数，第个正整数a；代表第i个斑点蛇第i厘米开始时有a；个斑点（1≤ai≤50）。
接下来每行有一条命令，命令有4种形式：
（1）Addij，i和j为正整数，表示第i厘米增加j个斑点（j不超过30）；
（2）Subij，i和i为正整数，表示第i厘米减少j个斑点（不超过30）；
（3）Queryij，i和j为正整数，i≤j，表示询问第i到第j厘米的斑点总数（包括第i厘米和第j厘米）；
（4）End 表示结束，这条命令在每组数据最后出现；

输出格式
对于每个Query询问，输出一个整数并回车，表示询问的段中的总斑点数，这个数保证在int 范围内。
样例输入
10
1 2 3 4 5 6 7 8 9 1 0
Query 1 3
Add 3 6
Query 2 7
Sub 10 2
Add 6 3
Query 3 10
End

样例输出
6
33
59

AC代码:线段树

区间求和

#include<iostream>
using namespace std;

int MAX_N = 50000;
int s[200005];

void up(int p){
    s[p] = s[p * 2] + s[p * 2 + 1];
}

void modify(int p,int l,int r,int x,int v,int pm){
    if(l == r){
        if(pm == 1){
            s[p] += v;
        }
        else{
            s[p] -= v;
        }

        return;
    }

    int mid = (l + r) / 2;
    if(x <= mid){
        modify(p * 2,l,mid,x,v,pm);
    }
    else{
        modify(p * 2 + 1,mid + 1,r,x,v,pm);
    }
    up(p);
}

int query(int p,int l,int r,int x,int y){
    if(x <= l && r <= y){
        return s[p];
    }
    int mid = (l + r) / 2;
    int res = 0;
    if(x <= mid){
        res += query(p * 2,l,mid,x,y);
    }
    if(y > mid){
        res += query(p * 2 + 1,mid + 1,r,x,y);
    }
    return res;
}

int main(){
    int n;
    cin>>n;
    for(int i=1;i<=n;++i){
        int x_temp;
        cin>>x_temp;
        modify(1,1,n,i,x_temp,1);
    }
    string type;
    cin>>type;
    while( "End" != type ){
        int x,y;
        cin>>x>>y;
        if( "Query" == type ){
            cout<<query(1,1,n,x,y)<<endl;
        }
        else if( "Add" == type ){
            modify(1,1,n,x,y,1);
        }
        else{
            modify(1,1,n,x,y,0);
        }
        cin>>type;
    }
    return 0;
}