UVALA 2191 Potentiometers 线段树

点击打开题目链接

线段树入门，模板

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;

#define Lson 2 * o, L, M        //左儿子，这样写减少了代码量，不易出错
#define Rson 2 * o + 1, M + 1, R    //ps：开始没有这样写，debug了好久

const int MAXM = 4 * 200000 + 10;   //结点总数不超过区间长度的四倍
int sumV[MAXM];
int ql, qr, p, v;
int n, m;
//向上更新，由 o 的左右儿子更新 o
void PushUp(int o)
{
    sumV[o] = sumV[2 * o] + sumV[2 * o + 1];
}
//建树
void Build(int o, int L, int R)
{
    if (L == R) scanf("%d", &sumV[o]);   //叶子结点
    else
    {
        int M = (L + R) / 2;
        Build(Lson);            //左子树
        Build(Rson);            //右子树
        PushUp(o);              //更新
    }
}
//查询区间ql，qr和 o为当前结点编号，L为当前结点左端点，R为右端点
int Query(int o, int L, int R)
{
    if (ql <= L && qr >= R) return sumV[o];  //当前结点完全包含在查询区间内
    int M = (L + R) / 2;
    int ans = 0;
    if (ql <= M) ans += Query(Lson);   //往左走，左儿子
    if (qr > M) ans += Query(Rson);    //往右走，右儿子
    return ans;
}

//修改A[p] = v，这里A为虚拟数组，我们已经将它保存在sumV中
void Update(int o, int L, int R)
{
    int M = (L + R) / 2;
    if (L == R) sumV[o] = v;	//叶结点，直接修改
    else
    {
        if (p <= M) Update(Lson);   //先递归更新左子树或者右子树
        else Update(Rson);
        PushUp(o);                  //然后计算本结点的sumV
    }
}

int main()
{
	int kase = 1;
	char ch[10];
	while (~scanf("%d", &n), n)
	{
		memset(sumV, 0, sizeof(sumV));
        Build(1, 1, n);
		
		if (kase != 1) printf("\n");
		printf("Case %d:\n", kase++);
		while (~scanf("%s", ch))
		{
			if (ch[0] == 'E') break;
			if (ch[0] == 'M')
			{
				scanf("%d%d", &ql, &qr);
				printf("%d\n", Query(1, 1, n));
			}
			else 
			{	
				scanf("%d%d", &p, &v);
				Update(1, 1, n);
			}
		}
	}
    return 0;
}