码迷,mamicode.com
首页 > 其他好文 > 详细

Hdu1166单点更新线段树

时间:2014-08-22 00:05:45      阅读:247      评论:0      收藏:0      [点我收藏+]

标签:style   blog   color   os   io   for   ar   div   

 

 

入门线段树,单点更新。写了几遍,都是学着notonlysuccess写的。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <climits>
#include <string>
#include <iostream>
#include <map>
#include <cstdlib>
#include <list>
#include <set>
#include <queue>
#include <stack>

using namespace std;
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
int sum[444444];
void up(int rt)
{
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
}

void build(int l, int r, int rt)
{
    if (l == r){
        scanf("%d", &sum[rt]); return;
    }
    int mid = (l + r) >> 1;
    build(lson);
    build(rson);
    up(rt);
}

void update(int key, int add, int l, int r, int rt)
{
    if (l == r){
        sum[rt] += add; return;
    }
    int mid = (l + r) >> 1;
    if (key <= mid) update(key, add, lson);
    else update(key, add, rson);
    up(rt);
}

int ask(int L, int R, int l, int r, int rt)
{
    if (L <= l&&r <= R) return sum[rt];
    int mid = (l + r) >> 1;
    int ans = 0;
    if (L <= mid) ans += ask(L, R, lson);
    if (R>mid) ans += ask(L, R, rson);
    return ans;
}
char str[100];
int main()
{
    int Icase; int n;
    scanf("%d", &Icase);
    int tt = 0;
    while (Icase--){
        printf("Case %d:\n", ++tt);
        scanf("%d", &n);
        build(1, n, 1);
        for (;;){
            int a; int b;
            scanf("%s", str); if (str[0] == E) break; scanf("%d%d", &a, &b);
            if (str[0] == A){
                update(a, b, 1, n, 1);
            }
            if (str[0] == S){
                update(a, -b, 1, n, 1);
            }
            if (str[0] == Q){
                printf("%d\n", ask(a, b, 1, n, 1));
            }
        }
    }
    return 0;
}

 

Hdu1166单点更新线段树,布布扣,bubuko.com

Hdu1166单点更新线段树

标签:style   blog   color   os   io   for   ar   div   

原文地址:http://www.cnblogs.com/yigexigua/p/3928225.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!