bzoj 3083: 遥远的国度

题意:

维护一棵树,支持换根,路径修改,当前根下的子树最小值。

题解:

首先可以发现换根没什么太大的用处,无论换了几次,只有最后一次是对当前有影响的。所以记录下现在哪个做根,树的形态不用动。
第二种操作直接树剖。
关键是查询,因为换根,有点麻烦。
假设整棵树一开始的根是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);
        }
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值