洛谷 P2173 [ZJOI2012]网络 —— LCT单点更新,RMQ,判连通

题意:

This way

题解:

有一个坑点就是改变边的颜色前后它的颜色可能是一样的。
建立10个LCT
首先第一个操作就是直接改x位置的值,但是要先access,因为我们要push_up,所以要连一条它到根的路径。但是就算不连,之后再splay一下也是对的?不太懂了。
第二个操作,我们直接用一个unordered_map来存即可。注意判相同。
对于每个点连边的数量我们直接用一个数组记录即可。
最后的话我们先判是否在一个连通块内,再split一下找答案即可。

#include<bits/stdc++.h>
#define G if(++ip==ie)if(fread(ip=buf,1,SZ,stdin))
#pragma GCC optimize(2)
#define ri register int
#define iv inline void
using namespace std;
const int N=1e5+5;
struct LCT
{
    int f[N],ch[N][2],rev[N],siz[N],num[N];
    inline bool is_rt(ri x){return ch[f[x]][0]==x||ch[f[x]][1]==x;}
    int a[N],mx[N];
    iv push_up(ri x)
    {
        siz[x]=siz[ch[x][0]]+siz[ch[x][1]]+1;
        mx[x]=max(a[x],max(mx[ch[x][0]],mx[ch[x][1]]));
    }
    iv reverse(ri x)
    {
        swap(ch[x][0],ch[x][1]);
        rev[x]^=1;
    }
    iv push_down(ri x)
    {
        if(!rev[x])
            return ;
        if(ch[x][0])
            reverse(ch[x][0]);
        if(ch[x][1])
            reverse(ch[x][1]);
        rev[x]=0;
    }
    iv rotate(ri x){//一次旋转
        int y=f[x],z=f[y],k=ch[y][1]==x,w=ch[x][!k];
        if(is_rt(y))ch[z][ch[z][1]==y]=x;ch[x][!k]=y;ch[y][k]=w;//额外注意if(nroot(y))语句,此处不判断会引起致命错误(与普通Splay的区别2)
        if(w)f[w]=y;f[y]=x;f[x]=z;
        push_up(y);
    }
    int st[N];
    iv splay(ri x){//只传了一个参数,因为所有操作的目标都是该Splay的根(与普通Splay的区别3)
        ri y=x,z=0;
        st[++z]=y;//st为栈,暂存当前点到根的整条路径,pushdown时一定要从上往下放标记(与普通Splay的区别4)
        while(is_rt(y))st[++z]=y=f[y];
        while(z)push_down(st[z--]);
        while(is_rt(x)){
            y=f[x];z=f[y];
            if(is_rt(y))
                rotate((ch[y][0]==x)^(ch[z][0]==y)?x:y);
            rotate(x);
        }
        push_up(x);
    }
    /*当然了,其实利用函数堆栈也很方便,代替上面的手工栈,就像这样
    I pushall(R x){
        if(is_rt(x))pushall(f[x]);
        pushdown(x);
    }*/
    iv access(ri x){//访问
        for(ri y=0;x;x=f[y=x])
            splay(x),ch[x][1]=y,push_up(x);
    }
    iv makeroot(ri x){//换根
        access(x);splay(x);
        reverse(x);
    }
    inline int findroot(ri x){//找根(在真实的树中的)
        access(x);splay(x);
        while(ch[x][0])push_down(x),x=ch[x][0];
        splay(x);
        return x;
    }
    iv split(ri x,ri y){//提取路径
        makeroot(x);
        access(y);
        splay(y);
    }
    iv link(ri x,ri y){//连边
        makeroot(x);
        f[x]=y;
        push_up(x);
    }
    iv cut(ri x,ri y){//断边
        makeroot(x);
        if(findroot(y)==x&&f[y]==x&&!ch[y][0]){
            f[y]=ch[x][1]=0;
            push_up(x);
        }
    }
}lct[11];
unordered_map<int,int>mp[N];
int main()
{
    int n,m,c,k;
    scanf("%d%d%d%d",&n,&m,&c,&k);
    for(int i=1;i<=n;i++)scanf("%d",&lct[1].a[i]),lct[1].mx[i]=lct[1].a[i];
    for(int i=2;i<=c;i++)
        for(int j=1;j<=n;j++)
            lct[i].a[j]=lct[i].mx[j]=lct[1].a[j];
    int x,y,col;
    for(int i=1;i<=m;i++)
        scanf("%d%d%d",&x,&y,&col),col++,lct[col].link(x,y),lct[col].num[x]++,lct[col].num[y]++,mp[x][y]=mp[y][x]=col;
    while(k--)
    {
        int op,x,y,w;
        scanf("%d",&op);
        if(op==0)
        {
            scanf("%d%d",&x,&y);
            for(int i=1;i<=c;i++)
            {
                lct[i].access(x);
                lct[i].a[x]=y;
                lct[i].push_up(x);
            }
        }
        else if(op==1)
        {
            scanf("%d%d%d",&x,&y,&w);
            w++;
            if(!mp[x][y])
            {
                printf("No such edge.\n");
                continue;
            }
            if(mp[x][y]==w)
            {
                printf("Success.\n");
                continue;
            }
            if(lct[w].num[x]>=2||lct[w].num[y]>=2)
            {
                printf("Error 1.\n");
                continue;
            }
            lct[w].access(y);
            if(lct[w].findroot(x)==lct[w].findroot(y))
                printf("Error 2.\n");
            else
            {
                col=mp[x][y];
                lct[col].cut(x,y),lct[w].link(x,y);
                lct[col].num[x]--,lct[col].num[y]--,lct[w].num[x]++,lct[w].num[y]++;
                mp[x][y]=mp[y][x]=w;
                printf("Success.\n");
            }
        }
        else
        {
            scanf("%d%d%d",&w,&x,&y);
            w++;
            lct[w].access(y);
            if(lct[w].findroot(x)!=lct[w].findroot(y))
                printf("-1\n");
            else
                lct[w].split(x,y),printf("%d\n",lct[w].mx[y]);
        }
    }
    return 0;
}
发布了530 篇原创文章 · 获赞 31 · 访问量 5万+

猜你喜欢

转载自blog.csdn.net/tianyizhicheng/article/details/101555102