UOJ #274. 【清华集训2016】温暖会指引我们前行 [lct]

#274. 【清华集训2016】温暖会指引我们前行

题意比较巧妙

裸lct维护最大生成树

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
#define lc t[x].ch[0]
#define rc t[x].ch[1]
#define pa t[x].fa
const int N = 4e5+5, INF = 1e9+5;
inline int read(){
    char c=getchar(); int x=0,f=1;
    while(c<'0' || c>'9') {if(c=='-')f=-1; c=getchar();}
    while(c>='0' && c<='9') {x=x*10+c-'0'; c=getchar();}
    return x*f;
}

int n, m, val[N], id, u, v, tem, w;
struct edge{int u, v;} e[N];
char s[10];

namespace lct {
    struct meow{ int ch[2], fa, rev, sum, w, p; } t[N];
    inline int wh(int x) {return t[pa].ch[1] == x;}
    inline bool isr(int x) {return t[pa].ch[0] != x && t[pa].ch[1] != x;}
    inline void rever(int x) {t[x].rev ^= 1; swap(lc, rc);}
    inline void pushdn(int x) {
        if(t[x].rev) {
            if(lc) rever(lc);
            if(rc) rever(rc);
            t[x].rev = 0;
        }
    }
    inline void pd(int x) {if(!isr(x)) pd(pa); pushdn(x);}
    inline void update(int x) {
        t[x].sum = t[lc].sum + t[rc].sum + t[x].w;
        t[x].p = x;
        if(lc && val[ t[lc].p ] < val[ t[x].p ]) t[x].p = t[lc].p;
        if(rc && val[ t[rc].p ] < val[ t[x].p ]) t[x].p = t[rc].p;
    }
    inline void rotate(int x) {
        int f = t[x].fa, g = t[f].fa, c = wh(x);
        if(!isr(f)) t[g].ch[wh(f)] = x; t[x].fa = g;
        t[f].ch[c] = t[x].ch[c^1]; t[ t[f].ch[c] ].fa = f;
        t[x].ch[c^1] = f; t[f].fa = x;
        update(f); update(x);
    }
    inline void splay(int x) {
        pd(x);
        for(; !isr(x); rotate(x))
            if(!isr(pa)) rotate(wh(x) == wh(pa) ? pa : x);
    }
    inline void access(int x) {
        for(int y=0; x; y=x, x=pa)
            splay(x), rc=y, update(x);
    }
    inline void maker(int x) { access(x); splay(x); rever(x);}
    inline void link(int x, int y) { maker(x); t[x].fa = y; }
    inline void cut(int x, int y) { 
        maker(x); access(y); splay(y); 
        t[x].fa = t[y].ch[0] = 0; update(y); 
    }
    inline void split(int x, int y) { maker(x), access(y); splay(y); }
} using namespace lct;

int fa[N];
inline int find(int x) {return x == fa[x] ? x : fa[x] = find(fa[x]);}
inline void add() {
    id=read()+1+n, u=read()+1, v=read()+1, tem=read(), w=read(); //printf("\nadd %d  %d--%d  %d  %d\n", id, u, v, tem, w);
    e[id] = (edge){u, v};
    val[id] = tem;
    t[id].sum = t[id].w = w; t[id].p = id;
    if(find(u) != find(v)) { 
        fa[find(u)] = find(v);
        link(id, u); link(id, v);
    } else {
        split(u, v);
        int a = t[v].p; //printf("aaa %d\n", a);
        if(val[a] < val[id]) { 
            cut(a, e[a].u); cut(a, e[a].v);
            link(id, u); link(id, v);
        }
    }
}
inline void que(int u, int v) {
    if(find(u) != find(v)) puts("-1");
    else split(u, v), printf("%d\n", t[v].sum);
}
inline void cha(int id, int w) { //printf("\ncha %d  %d\n", id, w);
    t[id].w = w; splay(id);
}
int main() {
    freopen("in", "r", stdin);
    n=read(); m=read();
    for(int i=1; i<=n; i++) fa[i] = i, val[i] = INF;
    for(int i=1; i<=m; i++) { //printf("\nQ %d\n", i);
        scanf("%s", s);
        if(s[0] == 'f') add();
        if(s[0] == 'm') u=read()+1, v=read()+1, que(u, v);
        if(s[0] == 'c') id=read()+1+n, w=read(), cha(id, w);
    }
}