BZOJ_P3651 網絡通信(LinkCutTree)

BZOJ傳送門

Time Limit: 10 Sec Memory Limit: 256 MB
Submit: 71 Solved: 52
[Submit][Status][Discuss]
Description
有一個由M 條電纜連接的 N 個站點組成的網絡。爲了防止壟斷，由 C 個公司控制所有的電纜，規定任何公司不能控制連接同一個站點的兩條以上的電纜（可以控制兩條）。同時規定，每個公司不能有多餘的電纜，所謂的多餘，是指屬於同一個公司的電纜不能形成環。
在運作過程中，不同公司之間會進行電纜買賣。請你寫一個程序判斷買賣是否合法。

Input
輸入第一行有4個由空格隔開的整數 N，M，C和 T。N(1≤N≤ 8 000)表示站點數，M(0≤M≤100 000)表示連接站點的電纜數。C(1≤C≤ 100)表表示公司數量，T 表示電纜買賣次
數。後面有M行，每行三個整數Sj1,Sj2和Kj，表示連接站點Sj1和Sj2(1≤Sj1< Sj2 ≤ n)的電纜屬於Kj(1≤Kj≤C)公司擁有，任意兩個站點只有一條直接相連的電纜，輸入狀態合法。最後T(0≤T≤100 000)行，每行三個整數 Si1, Si2和 Ki,表示 Ki公司想購買站點Si1和Si2之間的電纜。

Output
輸出共 T行，表示處理的結果，有以下幾種可能的結果：
1、“No such cable.” 兩個站點間沒有電纜。
2、 “Already owned.” 電纜己經是 Ki 公司控制。
3、 “Forbidden: monopoly.” Ki 公司己經控制了兩條連接 Si1 或 Si2 的電纜。
4、 “Forbidden: redundant.” Ki 公司控制的線路會出現環。
5、 “Sold.” 可以買賣。

Sample Input
4 5 3 5
1 2 1
2 3 1
3 4 2
1 4 2
1 3 3
1 2 3
1 2 3
1 4 3
2 3 3
2 4 3

Sample Output
Sold.
Already owned.
Forbidden: monopoly.
Forbidden: redundant.
No such cable.

HINT

Source

Sol:
比較的水，一種顏色一顆LCT，可見寫結構的函數的優越感QuQ (其實並不用LCT)

#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
#define C 102
#define N 8005
#define M 100005
#define lc(o) ch[o][0]
#define rc(o) ch[o][1]
inline int in(int x=0,char ch=getchar(),int v=1){
    while(ch!='-'&&(ch>'9'||ch<'0')) ch=getchar();if(ch=='-') v=-1,ch=getchar();
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();return x*v;
}
struct LinkCutTree{
    int f[N],ch[N][2],rev[N];
    bool isrt(int o){return (!f[o])||(o!=lc(f[o])&&o!=rc(f[o]));}
    inline void Updata(int o){if(rev[o]){rev[o]^=1;swap(lc(o),rc(o));rev[lc(o)]^=1,rev[rc(o)]^=1;}}
    inline void Push(int o){if(!isrt(o)) Push(f[o]);Updata(o);}
    inline void Rot(int o){
        int p=f[o],k=f[p],d=lc(p)==o;
        if(!isrt(p)) ch[k][rc(k)==p]=o;
        f[ch[o][d]]=p;f[o]=k;
        ch[p][d^1]=ch[o][d],ch[o][d]=p,f[p]=o;
    }
    inline void Splay(int o){
        int p,k;Push(o);
        while(!isrt(o)){
            p=f[o],k=f[p];
            if(isrt(p)) Rot(o);
            else if((rc(p)==o)==(rc(k)==p)) Rot(p),Rot(o);
            else Rot(o),Rot(o);
        }
    }
    inline void Access(int o,int x=0){for(;o;o=f[x=o]) Splay(o),rc(o)=x;}
    inline void Makert(int o){Access(o),Splay(o),rev[o]^=1;}
    inline void Link(int u,int v){Makert(u),f[u]=v;}
    inline void Cut(int u,int v){Makert(u),Access(v),Splay(v),f[lc(v)]=0,lc(v)=0;}
    inline bool Check1(int u,int v){
        Makert(u),Access(v),Splay(v);
        for(;u;u=f[u]) if(u==v) return 0;return 1;}
}c[C];

int n,m,cr,t,cnt;short s[N][C];
struct Edge{int u,v,w;}edge[M];vector<int> g[N];
inline void Add_Edge(int u,int v,int w){
    edge[++cnt]=(Edge){u,v,w};s[u][w]++,s[v][w]++;
    g[u].push_back(cnt),g[v].push_back(cnt);
}

inline int Check(int u,int v){
    if(g[u].size()>g[v].size()) swap(u,v);
    for(int i=0,lim=g[u].size();i<lim;i++) if(edge[g[u][i]].u==v||edge[g[u][i]].v==v) return g[u][i];
    return -1;
}
void solve(){
    n=in(),m=in(),cr=in(),t=in();int u,v,w,w1;
    for(int i=1;i<=m;i++){
        u=in(),v=in(),w=in();
        Add_Edge(u,v,w);c[w].Link(u,v);
    }
    while(t--){
        u=in(),v=in(),w=in();
        w1=Check(u,v);
        if(w1==-1){puts("No such cable.");continue;}
        if(edge[w1].w==w){puts("Already owned.");continue;}
        if(s[u][w]>1||s[v][w]>1){puts("Forbidden: monopoly.");continue;}
        if(!c[w].Check1(u,v)){puts("Forbidden: redundant.");continue;};
        c[edge[w1].w].Cut(u,v);c[w].Link(u,v);puts("Sold.");
        s[u][edge[w1].w]--,s[v][edge[w1].w]--;
        s[u][w]++,s[v][w]++,edge[w1].w=w;
    }
}
int main(){
    solve();
    return 0;
}