题意:
维护一棵树,支持换根,路径修改,当前根下的子树最小值。
题解:
首先可以发现换根没什么太大的用处,无论换了几次,只有最后一次是对当前有影响的。所以记录下现在哪个做根,树的形态不用动。
第二种操作直接树剖。
关键是查询,因为换根,有点麻烦。
假设整棵树一开始的根是1,那么当x做根时,只对x到1的路径上的点的子树产生影响。
那么其他点就直接dfs序化成区间求最值。
当一个点y是x的祖宗,那么它的子树就变成这样:
右上角为y的子树。
所以找到y到x的路径的第一个点,除去这棵子树其他就区间的最值就是答案。
code:
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;
struct trnode{
int fa,dep,son,tot,top,lc,rc,c,la,u;
}tr[200010];int ys[100010],z=0,tot=0,rt;
struct node{
int y,next;
}a[200010];int len=0,last[100010];
int n,m;
int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void ins(int x,int y)
{
a[++len].y=y;
a[len].next=last[x];last[x]=len;
}
void pre_node(int x,int fa)
{
tr[x].dep=tr[fa].dep+1;tr[x].fa=fa;tr[x].tot=1;
for(int i=last[x];i;i=a[i].next)
{
int y=a[i].y;
if(y==fa) continue;
pre_node(y,x);
if(tr[tr[x].son].tot<tr[y].tot) tr[x].son=y;
tr[x].tot+=tr[y].tot;
}
}
void pre_edge(int x,int top)
{
tr[x].top=top;ys[x]=tr[x].la=++z;
if(tr[x].son!=0) pre_edge(tr[x].son,top),tr[x].la=tr[tr[x].son].la;
for(int i=last[x];i;i=a[i].next)
{
int y=a[i].y;
if(y==tr[x].fa||y==tr[x].son) continue;
pre_edge(y,y);
tr[x].la=tr[y].la;
}
}
int bt(int l,int r)
{
int x=++tot;
if(l!=r)
{
int mid=(l+r)/2;
tr[x].lc=bt(l,mid);
tr[x].rc=bt(mid+1,r);
}
return x;
}
void update(int x)
{
int lc=tr[x].lc,rc=tr[x].rc,c=tr[x].u;
tr[lc].c=tr[rc].c=c;
tr[lc].u=tr[rc].u=c;
tr[x].u=0;
}
void change(int x,int l,int r,int fl,int fr,int c)
{
if(l==fl&&r==fr){tr[x].c=tr[x].u=c;return;}
int mid=(l+r)/2;
if(tr[x].u!=0) update(x);
if(fr<=mid) change(tr[x].lc,l,mid,fl,fr,c);
else if(fl>mid) change(tr[x].rc,mid+1,r,fl,fr,c);
else change(tr[x].lc,l,mid,fl,mid,c),change(tr[x].rc,mid+1,r,mid+1,fr,c);
tr[x].c=min(tr[tr[x].lc].c,tr[tr[x].rc].c);
}
int findans(int x,int l,int r,int fl,int fr)
{
if(l==fl&&r==fr) return tr[x].c;
int mid=(l+r)/2;
if(tr[x].u!=0) update(x);
if(fr<=mid) return findans(tr[x].lc,l,mid,fl,fr);
if(fl>mid) return findans(tr[x].rc,mid+1,r,fl,fr);
return min(findans(tr[x].lc,l,mid,fl,mid),findans(tr[x].rc,mid+1,r,mid+1,fr));
}
void solve(int x,int y,int c)
{
int tx=tr[x].top,ty=tr[y].top;
while(tx!=ty)
{
if(tr[tx].dep<tr[ty].dep) swap(x,y),swap(tx,ty);
change(1,1,n,ys[tx],ys[x],c);
x=tr[tx].fa,tx=tr[x].top;
}
if(tr[x].dep<tr[y].dep) swap(x,y);
change(1,1,n,ys[y],ys[x],c);
}
bool check(int x,int y)//判断x是否为y的祖宗
{
if(tr[x].dep>tr[y].dep) return false;
int ty=tr[y].top;
while(tr[ty].dep>tr[x].dep)
y=tr[ty].fa,ty=tr[y].top;
return tr[x].top==ty;
}
void work(int x)
{
if(rt==1||!check(x,rt)){printf("%d\n",findans(1,1,n,ys[x],tr[x].la));return;}
if(x==rt){printf("%d\n",findans(1,1,n,1,n));return;}
int y=rt,ty=tr[y].top,tx=tr[x].top;
while(ty!=tx&&tr[ty].fa!=x)
y=tr[ty].fa,ty=tr[y].top;
if(tx==ty) ty=tr[x].son;
int l=ys[ty],r=tr[ty].la;
int ans=(1<<28);
if(l!=1) ans=min(ans,findans(1,1,n,1,l-1));
if(r!=n) ans=min(ans,findans(1,1,n,r+1,n));
printf("%d\n",ans);
}
int main()
{
n=read();m=read();
for(int i=1;i<n;i++)
{
int x,y;x=read();y=read();
ins(x,y);ins(y,x);
}
pre_node(1,0);pre_edge(1,1);
bt(1,n);
for(int i=1;i<=n;i++)
{
int c;c=read();
change(1,1,n,ys[i],ys[i],c);
}
rt=read();
while(m--)
{
int opt;opt=read();
if(opt==1) rt=read();
if(opt==2)
{
int x,y,c;x=read();y=read();c=read();
solve(x,y,c);
}
if(opt==3)
{
int x;x=read();
work(x);
}
}
}