题意:
题解:
有一个坑点就是改变边的颜色前后它的颜色可能是一样的。
建立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;
}