您的位置:首页 > 其它

poj 3468__A Simple Problem with Integers(区间求和,成段更新)

2015-08-14 10:34 441 查看
链接:A Simple Problem with Integers

线段树的区间求和,成段更新模板题

代码如下

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define lson l , m , rt << 1
#define rson m+1 , r , rt << 1 | 1
#define ll long long
const int maxn=100000+10;
ll sum[maxn << 2], add[maxn << 2];
void Pushup(int rt)
{
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
}
void Pushdown(int rt, int m)    //m表示rt节点代表的区间长度
{
    if(add[rt]) {
        add[rt << 1] += add[rt];
        add[rt << 1 | 1] += add[rt];
        sum[rt << 1] += add[rt] * (m - (m >> 1));   //(m - (m >> 1))表示rt的左孩子增加(m - (m >> 1))个add[rt];
        sum[rt << 1 | 1] += add[rt] * (m >> 1);
        add[rt] = 0;    //处理完之后重置
    }
}
void build(int l, int r, int rt)
{
    add[rt] = 0;
    if(l == r) {
        scanf("%lld", &sum[rt]);
        return ;
    }
    int m = (r + l) >> 1;
    build(lson);
    build(rson);
    Pushup(rt);
}
ll query(int L, int R, int l, int r, int rt)
{
    if(L <= l && R >= r) {
        return sum[rt];
    }
    Pushdown(rt, r - l + 1);
    int m = (r + l) >> 1;
    ll ret = 0;
    if(L <= m) {
        ret += query(L, R, lson);
    }
    if(R > m) {
        ret += query(L, R, rson);
    }
    return ret;
}
void update(int L, int R, int c, int l, int r, int rt)
{
    if(L <= l && R >= r) {
        add[rt] += c;
        sum[rt] += (ll)c * (r - l + 1);
        return ;
    }
    Pushdown(rt, r - l + 1);
    int m = (l + r) >> 1;
    if(L <= m)
        update(L, R, c, lson);
    if(m < R)
        update(L, R, c, rson);
    Pushup(rt);
}
int main()
{
    int n, q, a, b, c;
    char s[2];
    scanf("%d%d", &n, &q);  //n个数,q次询问
    build(1, n, 1);
    while(q--) {
        scanf("%s", s);
        if(s[0] == 'Q') {
            scanf("%d%d", &a, &b);
            printf("%lld\n", query(a, b, 1, n, 1));
        } else if(s[0] == 'C') {
            scanf("%d%d%d", &a, &b, &c);
            update(a, b, c, 1, n, 1);
        }
    }
    return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: