sdut 4295 种西瓜 （2018西安邀请赛K）【线段树】

种西瓜

Time Limit: 1000 ms Memory Limit: 65536 KiB

Problem Description

现在郭小冉决定在瓜场种m年西瓜。
首先他有n块西瓜地。
第i块西瓜地每年可以将会生产出b（i）颗西瓜。
每一年ta都会回到西瓜地。
每一年ta有两种操作，一种是选择一段连续的土地查询它们这些年的产量总和。
另外一种是选择一段连续的土地，并把它们的每一块的年产量都增加1.

Input

每组输入包含多组测试数据
对于每组测试数据
第一行包含西瓜地的块数n。
第二行包含n块西瓜地的年产量。
第三行包含m 代表接下来的年数。
每一年都会有一个操作
查询格式如下：
查询产量 Q s t (1<=s,t<=n)
增加年产量 I s t (1<=s,t<=n)
只有在查询的时候进行输出。

1<=n<=100000

1<=m<=100000

1<=b(i)<=100000

Output

对于每组测试，输出一行由空格隔开的整数（个数为输入中查询Q的个数）

Sample Input

4
1 1 1 1
3
Q 1 3
Q 1 4
Q 2 4
4
1 2 3 4 
3
Q 1 4
I 1 3
Q 2 4

Sample Output

3 8 9
10 29

建立两课线段树， A树维护更改后的每年产量，B树维护每个位置是从哪年开始产量增加，所以查询操作即可转换为：

Query(A) * i - Query(B);

具体实现请结合代码食用；

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define lson rt << 1, l, mid
#define rson rt << 1 | 1, mid + 1, r
const int MAX = 1e5 + 7;
const int INF = 0x3f3f3f3f;
ll a[MAX << 2], lz[MAX << 2], b[MAX << 2];
ll lzb[MAX << 2];
void pushup(int rt){
    a[rt] = a[rt << 1] + a[rt << 1 | 1];
}
void pushupb(int rt){
    b[rt] = b[rt << 1] + b[rt << 1 | 1];
}
void pushdown(int rt, int l, int r){
    if(lz[rt]){
        int mid = (l + r) >> 1;
        lz[rt << 1] += lz[rt];
        lz[rt << 1 | 1] += lz[rt];
        a[rt << 1] += lz[rt] * (mid - l + 1);
        a[rt << 1 | 1] += (r - mid) * lz[rt];
        lz[rt] = 0;
    }
}
void pushdownb(int rt, int l, int r){
    if(lzb[rt]){
        int mid = (l + r) >> 1;
        lzb[rt << 1] += lzb[rt];
        lzb[rt << 1 | 1] += lzb[rt];
        b[rt << 1] += lzb[rt] * (mid - l + 1);
        b[rt << 1 | 1] += (r - mid) * lzb[rt];
        lzb[rt] = 0;
    }
}
void build(int rt, int l, int r){
    if(l == r){
        scanf("%lld", &a[rt]);
        return;
    }
    int mid = (l + r) >> 1;
    build(lson);
    build(rson);
    pushup(rt);
}
void update(int rt, int l, int r, int x, int y){
    if(x <= l && r <= y){
        lz[rt] += 1;
        a[rt] += (ll)(r - l + 1) * 1ll;
        return;
    }
    pushdown(rt, l, r);
    int mid = (l + r) >> 1;
    if(x <= mid) update(lson, x, y);
    if(mid < y) update(rson, x, y);
    pushup(rt);
}
void updateb(int rt, int l, int r, int x, int y, int v){
    if(x <= l && r <= y){
        lzb[rt] += v;
        b[rt] += (ll)v * (r - l + 1);
        return;
    }
    pushdownb(rt, l, r);
    int mid = (l + r) >> 1;
    if(x <= mid) updateb(lson, x, y, v);
    if(mid < y) updateb(rson, x, y, v);
    pushupb(rt);
}
ll query(int rt, int l, int r, int x, int y){
    if(x <= l && r <= y) return a[rt];
    pushdown(rt, l, r);
    int mid = (l + r) >> 1;
    ll ans = 0;
    if(x <= mid) ans += query(lson, x, y);
    if(mid < y) ans += query(rson, x, y);
    return ans;
}
ll queryb(int rt, int l, int r, int x, int y){
    if(x <= l && r <= y) return b[rt];
    pushdownb(rt, l, r);
    int mid = (l + r) >> 1;
    ll ans = 0;
    if(x <= mid) ans += queryb(lson, x, y);
    if(mid < y) ans += queryb(rson, x, y);
    return ans;
}
int main(){
    int n, m;
    while(scanf("%d", &n) != EOF){
        memset(lz, 0, sizeof lz);
        memset(b, 0, sizeof b);
        memset(lzb, 0, sizeof lzb);
        build(1, 1, n);
        scanf("%d", &m);
        vector <ll> v;
        for(int i = 1; i <= m; i++){
            int a, b;
            char s[10];
            scanf("%s", s);
            scanf("%d%d", &a, &b);
            if(s[0] == 'Q'){
                ll ans = query(1, 1, n, a, b) * i - queryb(1, 1, n, a, b);
                v.push_back(ans);
            }
            if(s[0] == 'I'){
                update(1, 1, n, a, b);
                updateb(1, 1, n, a, b, i);
            }
        }
        for(int i = 0; i < v.size(); i++)
            printf("%lld%c", v[i], i == v.size() - 1 ? '\n' : ' ');
    }
    return 0;
}


/***************************************************
User name: headhard
Result: Accepted
Take time: 688ms
Take Memory: 1028KB
Submit time: 2018-08-03 13:41:03
****************************************************/