POJ 3468 A Simple Problem with Integers (线段树 区间增减 区间求和)

题目大意:

中文题面, 就是成段加减, 询问区间和


大致思路:

练习一下懒惰标记...就是一个简单的懒惰标记向下递推


代码如下:

Result  :  Accepted     Memory  :  4256 KB     Time  :  1782 ms

/*
 * Author: Gatevin
 * Created Time:  2015/8/16 19:53:07
 * File Name: Sakura_Chiyo.cpp
 */
#include<iostream>
#include<sstream>
#include<fstream>
#include<vector>
#include<list>
#include<deque>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<bitset>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cctype>
#include<cmath>
#include<ctime>
#include<iomanip>
using namespace std;
const double eps(1e-8);
typedef long long lint;

#define maxn 100010

struct Segment_Tree
{
    #define lson l, mid, rt << 1
    #define rson mid + 1, r, rt << 1 | 1
    lint val[maxn << 2];
    lint flag[maxn << 2];
    void pushUp(int rt)
    {
        val[rt] = val[rt << 1] + val[rt << 1 | 1];
        return;
    }
    void pushDown(int l, int r, int rt, int mid)
    {
        if(flag[rt])
        {
            flag[rt << 1] += flag[rt];
            flag[rt << 1 | 1] += flag[rt];
            val[rt << 1] += (mid - l + 1)*flag[rt];
            val[rt << 1 | 1] += (r - mid)*flag[rt];
            flag[rt] = 0;
        }
        return;
    }
    
    void build(int l, int r, int rt)
    {
        if(l == r)
        {
            scanf("%I64d", &val[rt]);
            flag[rt] = 0;
            return;
        }
        int mid = (l + r) >> 1;
        flag[rt] = 0;
        build(lson);
        build(rson);
        pushUp(rt);
    }
    void update(int l, int r, int rt, int L, int R, lint value)
    {
        if(l >= L && r <= R)
        {
            val[rt] += (r - l + 1)*value;
            flag[rt] += value;
            return;
        }
        int mid = (l + r) >> 1;
        pushDown(l, r, rt, mid);
        if(mid >= L) update(lson, L, R, value);
        if(mid + 1 <= R) update(rson, L, R, value);
        pushUp(rt);
    }
    lint query(int l, int r, int rt, int L, int R)
    {
        if(l >= L && r <= R)
            return val[rt];
        int mid = (l + r) >> 1;
        pushDown(l, r, rt, mid);
        lint ret = 0;
        if(mid >= L) ret += query(lson, L, R);
        if(mid + 1 <= R) ret += query(rson, L, R);
        return ret;
    }
};

Segment_Tree ST;

int main()
{
    int n, Q;
    while(~scanf("%d %d", &n, &Q))
    {
        ST.build(1, n, 1);
        char op[4];
        while(Q--)
        {
            scanf("%s", op);
            int a, b, c;
            if(op[0] == 'Q')
            {
                scanf("%d %d", &a, &b);
                printf("%I64d\n", ST.query(1, n, 1, a, b));
            }
            else
            {
                scanf("%d %d %d", &a, &b, &c);
                ST.update(1, n, 1, a, b, c);
            }
        }
    }
    return 0;
}


你可能感兴趣的:(线段树,poj,3468)