计蒜客习题 斑点蛇

问题描述

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

输入格式

第一行一个正整数 N（N≤50000）表示这条斑点蛇长度为 N 厘米，接下来有 N 个正整数，第 i 个正整数 ai​ 代表第 i 个斑点蛇第 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 厘米的斑点总数（包括第 i 厘米和第 j 厘米）；
(4) End 表示结束，这条命令在每组数据最后出现；
最多有 40000 条命令。

输出格式

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

分析

这道题就是计蒜客前面所描述的模板，直接把代码稍微一改就能AC了。

#include<bits/stdc++.h>
using namespace std;
const int MAX_N = 50010;
int s[4*MAX_N];
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)
{
    if(l==r)
    {
        s[p]+=v;
        return ;
    }
    int mid=(l+r)/2;
    if(x<=mid)
    {
        modify(p*2,l,mid,x,v);
    }
    else
    {
        modify(p*2+1,mid+1,r,x,v);
	}
    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,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;
    scanf("%d",&n);
    for(int i=1;i<=n;++i)
    {
        int d;
        scanf("%d",&d);
        modify(1,1,n,i,d);
    }
    char op;
    while((cin>>op)&&(op[0]!='E'))
    {
		int i,j;
    	scanf("%d%d",&i,&j);
		if(op[0]=='A')
		modify(1,1,n,i,j);
		else if(op[0]=='S')
		modify(1,1,n,i,-j);
		else
		printf("%d\n",query(1,1,n,i,j));
    }
    return 0;
}