hihocoder#1333 : 平衡树·Splay2 (区间操作）

题面：

#1333 : 平衡树·Splay2

时间限制:10000ms 单点时限:1000ms 内存限制:256MB

描述

小Ho：好麻烦啊~~~~~

小Hi：小Ho你在干嘛呢？

小Ho：我在干活啊！前几天老师让我帮忙管理一下团队的人员，但是感觉好难啊。

小Hi：说来听听？

小Ho：事情是这样的。我们有一个运动同好会，每天都有人加入或者退出，所以老师让我帮忙管理一下人员。每个成员有一个互不相同的id和他对我们同好会的兴趣值val，每隔一段时间一些成员的兴趣值就会发生变化。老师有时候也会问我一些成员的兴趣值。

小Hi：所以你就需要一个表格来管理信息咯？

小Ho：是啊，但是我们同好会的成员实在是太多了！我感觉完全搞不定啊。

小Hi：这样啊，那不如让我来帮帮你吧！

小Ho：真的吗？

小Hi：当然是真的啦，小Ho，你先告诉我有多少种需要完成的事情？

小Ho：一共有4种情况：

1. 加入：一个新的成员加入同好会，我会分配给他一个没有使用的id，并且询问他的兴趣值val。

2. 修改：id在区间[a,b]内的成员，兴趣值同时改变k，k有可能是负数，表示他们失去了对同好会的兴趣。

3. 退出：id在区间[a,b]内的成员要退出同好会，虽说是区间，也有可能只有1个人。

4. 询问：老师会问我在区间[a,b]内的成员总的兴趣值。

小Hi：我明白了，让我想一想该如何解决。

提示：Splay

输入

第1行：1个正整数n，表示操作数量，100≤n≤200,000

第2..n+1行：可能包含下面4种规则：

1个字母'I'，紧接着2个数字id,val，表示一个编号为id的新成员加入，其兴趣值为val，1≤id≤100,000,000，1≤val≤10,000,000，保证在团队中的每个人id都不相同。

1个字母'Q'，紧接着2个数字a,b。表示询问团队中id在区间[a,b]的所有成员总兴趣值，保证区间内至少有一个成员，结果有可能超过int的范围。

1个字母'M'，紧接着3个数字a,b,d，表示将团队中id在区间[a,b]的成员兴趣值都改变d，其中d有可能为负数。保证操作之后每个成员的兴趣值仍然在0~10,000,000。

1个字母'D'，紧接着2个数字a,b，表示将团队中id在区间[a,b]的成员除去。

注意有可能出现一个id为1的成员加入团队，被除去之后，又有一个新的id为1的成员加入团队的情况。

输出

若干行：每行1个整数，表示针对询问的回答，保证一定有合法的解

样例输入

9
I 1 1
I 2 2
I 3 3
Q 1 3
M 1 2 2
Q 1 3
D 2 3
I 4 2
Q 1 4

样例输出

6
10
5


链接：http://hihocoder.com/problemset/problem/1333


代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll M = 2e5+10;
const ll inf = 0x3f3f3f;
ll n,sz,rt,t1,t2;
ll c[M][2],sum[M],fa[M],id[M],tag[M],v[M],siz[M];
inline void pushup(ll k){
    ll l = c[k][0],r = c[k][1];
    sum[k] = sum[l] + sum[r] + v[k];
    siz[k] = siz[l] + siz[r] + 1;
}

void pushdown(ll k){
    ll l = c[k][0],r = c[k][1],t = tag[k];
    if(t){
        tag[k] = 0;
        if(l) tag[l]+=t,sum[l]+=t*siz[l],v[l]+=t;
        if(r) tag[r]+=t,sum[r]+=t*siz[r],v[r]+=t;
    }
}

void rotate(ll x,ll &k){
    ll y = fa[x],z = fa[y],l,r;
    pushdown(z); pushdown(y); pushdown(x);
    if(c[y][0] == x) l = 0;
    else l = 1;
    r = l^1;
    if(y == k) k = x;
    else {
        if(c[z][0]==y) c[z][0]=x;
        else c[z][1] = x;
    }
    fa[x] = z;fa[y] = x;fa[c[x][r]]=y;
    c[y][l]=c[x][r]; c[x][r]=y;
    pushup(y); pushup(x);
}

void splay(ll x,ll &k){
    pushdown(k);
    while(x != k){
        ll y = fa[x],z = fa[y];
        if(y != k){
            if(c[y][0]==x^c[z][0]==y)rotate(x,k);
            else rotate(y,k);
        }
        rotate(x,k);
    }
}

inline void ask_befor(ll k,ll x){
    pushdown(k);
    if(k == 0) return ;
    if(id[k] <= x) {
        t1 = k;
        ask_befor(c[k][1],x);
    }
    else ask_befor(c[k][0],x);
}

inline void ask_after(ll k,ll x){
    pushdown(k);
    if(k == 0) return ;
    if(id[k] >= x) {
        t2 = k;
        ask_after(c[k][0],x);
    }
    else ask_after(c[k][1],x);
}

inline void ins(ll &k,ll idx,ll val,ll last){
    pushdown(k);
    if(k == 0){
        ++sz;
        siz[sz] = 1;
        k = sz;
        v[k] = val;
        sum[k] = val;
        fa[k] = last;
        id[k] = idx;
        return ;
    }
    if(idx < id[k]) ins(c[k][0],idx,val,k);
    else ins(c[k][1],idx,val,k);
    pushup(k);
}

inline void update(ll l,ll r,ll val){
    ask_befor(rt,l-1);ask_after(rt,r+1);
    ll x = t1,y = t2;
    splay(x,rt); splay(y,c[x][1]);
    ll z = c[y][0];
    tag[z] += val;v[z] += val; sum[z] += val*siz[z];
}

void del(ll l,ll r){
    ask_befor(rt,l-1);ask_after(rt,r+1);
    ll x = t1,y = t2;
    splay(x,rt); splay(y,c[x][1]);
    c[y][0] = 0;
}

inline void query(ll l,ll r){
    ask_befor(rt,l-1);ask_after(rt,r+1);
    ll x = t1,y = t2;
    splay(x,rt); splay(y,c[x][1]);
    ll z = c[y][0];
    printf("%lld\n",sum[z]);
}


int main()
{
    sz = 0;
    ins(rt,-1,0,0);
    ins(rt,100000001,0,rt);
    scanf("%lld",&n);
    for(ll i = 1;i <= n;i ++){
        ll l,r,val;
        string s;
        cin>>s;
        scanf("%lld%lld",&l,&r);
        if(s[0] == 'M') scanf("%lld",&val),update(l,r,val);
        else if(s[0] == 'I') ins(rt,l,r,rt),splay(sz,rt);
        else if(s[0] == 'Q') query(l,r);
        else del(l,r);
    }
    return 0;
}

 

标签：rt,val,ll,Splay2,hihocoder,1333,void,sum,id
来源： https://www.cnblogs.com/kls123/p/10707172.html