bzoj3083 遥远的国度

传送门
Description

描述
zcwwzdjn在追杀十分sb的zhx，而zhx逃入了一个遥远的国度。当zcwwzdjn准备进入遥远的国度继续追杀时，守护神RapiD阻拦了zcwwzdjn的去路，他需要zcwwzdjn完成任务后才能进入遥远的国度继续追杀。

问题是这样的：遥远的国度有n个城市，这些城市之间由一些路连接且这些城市构成了一颗树。这个国度有一个首都，我们可以把这个首都看做整棵树的根，但遥远的国度比较奇怪，首都是随时有可能变为另外一个城市的。遥远的国度的每个城市有一个防御值，有些时候RapiD会使得某两个城市之间的路径上的所有城市的防御值都变为某个值。RapiD想知道在某个时候，如果把首都看做整棵树的根的话，那么以某个城市为根的子树的所有城市的防御值最小是多少。由于RapiD无法解决这个问题，所以他拦住了zcwwzdjn希望他能帮忙。但zcwwzdjn还要追杀sb的zhx，所以这个重大的问题就被转交到了你的手上。

Input

第1行两个整数n m，代表城市个数和操作数。
第2行至第n行，每行两个整数 u v，代表城市u和城市v之间有一条路。
第n+1行，有n个整数，代表所有点的初始防御值。
第n+2行一个整数 id，代表初始的首都为id。
第n+3行至第n+m+2行，首先有一个整数opt，如果opt=1，接下来有一个整数id，代表把首都修改为id；如果opt=2，接下来有三个整数p1 p2 v，代表将p1 p2路径上的所有城市的防御值修改为v；如果opt=3，接下来有一个整数 id，代表询问以城市id为根的子树中的最小防御值。

Output

对于每个opt=3的操作，输出一行代表对应子树的最小点权值。

Sample Input

3 7

1 2

1 3

1 2 3

1

3 1

2 1 1 6

3 1

2 2 2 5

3 1

2 3 3 4

3 1

Sample Output

1

2

3

4

提示

对于20%的数据，n<=1000 m<=1000。

对于另外10%的数据，n<=100000，m<=100000，保证修改为单点修改。

对于另外10%的数据，n<=100000，m<=100000，保证树为一条链。

对于另外10%的数据，n<=100000，m<=100000，没有修改首都的操作。

对于100%的数据，n<=100000，m<=100000，0<所有权值<=2^31。
HINT

Source

zhonghaoxi提供

题解

这道题乍一看仅仅是一道树链剖分，但是读过题后发现这道题非常毒瘤，因为这道题需要支持换根。
换根？每次重新dfs、建立线段树不就好了么？
然而题目的范围是100000，显然是不行的。
那该怎么办呢？
我们可以想到，仅仅保留真实情况下根的位置，而事实上并不去动原树，在询问的时候按照真实的根来查询。查询时的情况分三类：
①根和当前点重合，直接输出整个线段树中的最小值。
②根和当前点的LCA就是当前点，也就是说根在当前点的子树中，这样的话我们只需要查询线段树中去除当前点包含根的儿子那一部分的最小值。（蒟蒻开始只会枚举所有可能的位置，两个点无限T飞QAQ）
③其他情况直接输出以当前点为根的子树的最小值。

其他的操作就是树剖的普通操作。

有一个小小的坑点：本题的数据范围是0~2^31-1，结果我已开始用的INF=1e9，WA了4个点

ps：我竟然没有发现一个人直接利用树剖性质来求LCA的，全都用的倍增，那让我来给各位科普一下吧。花式找LCA

在COGS上的提交，23333
CODE：

#include<cstdio>
const int INF=2147483647;
const int N=1e5+10;
struct edge
{
    int next,to;
}a[N<<1];
struct tree
{
    int num,change;
}t[N<<2];
int head[N],size[N],son[N],deep[N],pos[N],top[N],f[N],s[N],num[N];
int n,q,x,y,z,opt,e_num,tot,root;

inline int max(int a,int b){return a>b?a:b;}
inline int min(int a,int b){return a<b?a:b;}
inline void swap(int &a,int &b){a^=b,b^=a,a^=b;}
inline void read(int &n)
{
    n=0;char c=getchar();
    while(c<'0'||c>'9') c=getchar();
    while(c>='0'&&c<='9') n=n*10+c-48,c=getchar();
}
inline void add(int x,int y)
{
    a[++e_num].next=head[x],a[e_num].to=y,head[x]=e_num;
    a[++e_num].next=head[y],a[e_num].to=x,head[y]=e_num;
}
void dfs(int now,int fa,int depth)
{
    deep[now]=depth;
    f[now]=fa;
    size[now]=1;
    int tmp=-INF;
    for(int i=head[now];i;i=a[i].next)
      if(a[i].to!=fa)
      {
        dfs(a[i].to,now,depth+1);
        size[now]+=size[a[i].to];
        if(size[a[i].to]>tmp) tmp=size[a[i].to],son[now]=a[i].to;
      }
}
void dfs2(int now,int high)
{
    top[now]=high;
    pos[now]=++tot;
    num[tot]=s[now];
    if(son[now]) dfs2(son[now],high);
    for(int i=head[now];i;i=a[i].next)
      if(a[i].to!=f[now]&&a[i].to!=son[now]) dfs2(a[i].to,a[i].to);
}
inline void pushdown(int l,int r,int now)
{
    if(!t[now].change) return;
    int s1=now<<1,s2=s1|1;
    t[s1].num=t[s2].num=t[now].change;
    t[s1].change=t[s2].change=t[now].change;
    t[now].change=0;
}
inline void update(int now)
{
    t[now].num=min(t[now<<1].num,t[now<<1|1].num);
}
void build(int l,int r,int now)
{
    if(l==r)
    {
        t[now].num=num[l];
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,now<<1);
    build(mid+1,r,now<<1|1);
    update(now);
}
void Add(int L,int R,int l,int r,int now,int num)
{
    if(L<=l&&r<=R)
    {
        t[now].num=t[now].change=num;
        return;
    }
    int mid=(l+r)>>1;
    pushdown(l,r,now);
    if(L<=mid) Add(L,R,l,mid,now<<1,num);
    if(R>mid) Add(L,R,mid+1,r,now<<1|1,num);
    update(now);
}
int ask(int L,int R,int l,int r,int now)
{
    if(L<=l&&r<=R) return t[now].num;
    int mid=(l+r)>>1,ans=INF;
    pushdown(l,r,now);
    if(L<=mid) ans=min(ans,ask(L,R,l,mid,now<<1));
    if(R>mid) ans=min(ans,ask(L,R,mid+1,r,now<<1|1));
    return ans;
}
void addpath(int x,int y,int z)
{
    int L,R;
    if(top[x]==top[y])
    {
        L=min(pos[x],pos[y]);
        R=max(pos[x],pos[y]);
        return Add(L,R,1,n,1,z);
    }
    if(deep[top[x]]<deep[top[y]]) swap(x,y);
    L=min(pos[x],pos[top[x]]);
    R=max(pos[x],pos[top[x]]);
    Add(L,R,1,n,1,z);
    return addpath(y,f[top[x]],z);
}
inline int LCA(int x,int y)
{
    while(top[x]!=top[y])
      if(deep[top[x]]>deep[top[y]]) x=f[top[x]];
      else y=f[top[y]];
    return deep[x]<deep[y]?x:y;
}
int askone(int p,int l,int r,int now)
{
    if(l==r) return t[now].num;
    int mid=(l+r)>>1;
    pushdown(l,r,now);
    if(p<=mid) return askone(p,l,mid,now<<1);
    return askone(p,mid+1,r,now<<1|1);
}
inline int Ask(int x)
{
    if(root==x) return t[1].num;
    if(LCA(x,root)==x)
    {
        int ans=INF,from;
        for(int i=head[x];i;i=a[i].next)
          if(LCA(a[i].to,root)==a[i].to){from=a[i].to;break;}
        if(pos[from]>1) ans=min(ans,ask(1,pos[from]-1,1,n,1));
        if(pos[from]+size[from]<=n) ans=min(ans,ask(pos[from]+size[from],n,1,n,1));
        return ans;
    }
    return ask(pos[x],pos[x]+size[x]-1,1,n,1);
}
int main()
{
    read(n),read(q);
    for(int i=1;i<n;i++)
      read(x),read(y),add(x,y);
    for(int i=1;i<=n;i++)
      read(s[i]);
    read(root);
    dfs(1,0,1);
    dfs2(1,1);
    build(1,n,1);
    while(q--)
    {
        read(opt);
        if(opt==1) read(root);
        else if(opt==2) read(x),read(y),read(z),addpath(x,y,z);
        else read(x),printf("%d\n",Ask(x));
    }
    return 0;
}

UPD：原先的swap函数是错的，现已更正，希望大家不要被我这个zz误导

原本枚举所有情况的丑陋代码QAQ：

inline int Ask(int x)
{
    if(root==x) return t[1].num;
    if(LCA(x,root)==x)
    {
        int ans=INF,from;
        for(int i=head[x];i;i=a[i].next)
          if(LCA(a[i].to,root)==a[i].to){from=a[i].to;break;}
        if(pos[from]>1) ans=min(ans,ask(1,pos[from]-1,1,n,1));
        if(pos[from]+size[from]<=n) ans=min(ans,ask(pos[from]+size[from],n,1,n,1));
//      for(int i=head[x];i;i=a[i].next)
//        if(LCA(a[i].to,root)==a[i].to){from=a[i].to;break;}
//      while(x)
//      {
//          ans=min(ans,askone(pos[x],1,n,1));
//          for(int i=head[x];i;i=a[i].next)
//            if(a[i].to!=last&&a[i].to!=from&&a[i].to!=f[x]) ans=min(ans,ask(pos[a[i].to],pos[a[i].to]+size[a[i].to]-1,1,n,1));
//          last=x;x=f[x];
//      }
        return ans;
    }
    return ask(pos[x],pos[x]+size[x]-1,1,n,1);
}