线段树入门,模板
#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;
}