ACM-ICPC 2018 徐州赛区网络预赛 H. Ryuji doesn't want to study （线段树）

题目大意

给出一个序列，有两种操作一个是求一个区间

a[l]×L+a[l+1]×(L−1)+⋯+a[r−1]×2+a[r] ，L是区间[l, r]的长度。

另一个操作是单点修改。

思路

很明显是线段树问题，一开始并没有想到如何进行合并，也就是如何把两个区间合并成一个区间。不过我们可以线段树的一个节点存两个值一个是区间的和sum1另一个是区间的(n - i + 1)*a[i]的和sum2然后我们要求的就是sum2 - sum1×(n-r)

###代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxn = 1e5 + 100;

struct sc {
    int l, r;
    ll sum;
    ll val_sum;
};

int n;
sc tr[maxn << 2];
ll v[maxn];
ll sum, sumval;

void build(int l, int r, int k) {
    tr[k].l = l, tr[k].r = r;
    if(l == r) {
        tr[k].sum = v[l];
        tr[k].val_sum = v[l] * (n-l+1);
        return ;
    }
    int mid = (tr[k].l+tr[k].r)>>1;
    build(l,mid,k*2);
    build(mid+1,r,k*2+1);
    tr[k].sum = tr[k*2].sum + tr[k*2+1].sum;
    tr[k].val_sum = tr[k*2].val_sum + tr[k*2+1].val_sum;
}

void query(int l, int r, int k) {
    if(tr[k].r <= r && l <= tr[k].l) {
        sum += tr[k].sum;
        sumval += tr[k].val_sum;
        return ;
    }
    int mid = (tr[k].l+tr[k].r)>>1;
    if(r <= mid) {
        query(l, r, k*2);
    }
    else if(l > mid) {
        query(l, r, k*2+1);
    }
    else {
        query(l,mid,k*2);
        query(mid+1,r,k*2+1);
    }
}


void update(int idx, ll x, int k) {
    if(tr[k].l==tr[k].r && tr[k].r == idx) {
        tr[k].sum = x;
        tr[k].val_sum = x * (n - idx + 1);
        return ;
    }
    int mid = (tr[k].l+tr[k].r) / 2;
    if(idx <= mid) {
        update(idx, x, k*2);
    }
    else update(idx, x, k*2+1);
    tr[k].sum = tr[k*2].sum + tr[k*2+1].sum;
    tr[k].val_sum = tr[k*2].val_sum + tr[k*2+1].val_sum;
}

int main(){
    //freopen("input.txt", "r", stdin);
    int m;
    cin >> n >> m;
    for(int i = 1; i <= n; i++) {
        scanf("%lld", &v[i]);
    }
    build(1, n, 1);
    for(int k = 1; k <= m; k++) {
        int q;
        cin >> q;
        if(q==1) {
            int l, r;
            cin >> l >> r;
            sum = sumval = 0;
            query(l, r, 1);
            cout << sumval - sum * (n - r) << endl;
        }
        else {
            int idx;
            ll x;
            cin >> idx >> x;
            update(idx, x, 1);
        }
    }
    return 0;
}