UVALA 2191 Potentiometers 线段树

来源：互联网发布：无法打开awrrpt.sql 编辑：程序博客网时间：2024/06/13 01:20
点击打开题目链接
线段树入门，模板
#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 的左右儿子更新 ovoid 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;}
0 0