Description
给定一棵大小为 n 的有根点权树,支持以下操作:
• 换根
• 修改点权
• 查询子树最小值
Input
第一行两个整数 n, Q ,分别表示树的大小和操作数。
接下来n行,每行两个整数f,v,第i+1行的两个数表示点i的父亲和点i的权。保证f < i。如 果f = 0,那么i为根。输入数据保证只有i = 1时,f = 0。
接下来 m 行,为以下格式中的一种:
• V x y表示把点x的权改为y
• E x 表示把有根树的根改为点 x
• Q x 表示查询点 x 的子树最小值
Output
对于每个 Q ,输出子树最小值。
Sample Input
3 7
0 1
1 2
1 3
Q 1
V 1 6
Q 1
V 2 5
Q 1
V 3 4
Q 1
Sample Output
1
2
3
4
HINT
对于 100% 的数据:n, Q ≤ 10^5。
题解
以1为根建立dfs树,对于询问,考虑x与根的关系。
代码
#include<bits/stdc++.h>
#define inf 1000000000
#define mod 1000000007
#define ll long long
using namespace std;
inline int read()
{
int x=0;char ch=getchar();
while (ch<'0'||ch>'9') ch=getchar();
while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x;
}
int n,Q,tim,tot,rt;
int Head[100005],ret[200005],Next[200005];
int val[100005],l[100005],r[100005],q[100005],dep[100005];
int fa[100005][20],fac[20];
int mn[400005];
void ins(int u,int v){ret[++tot]=v;Next[tot]=Head[u];Head[u]=tot;}
void dfs(int u)
{
tim++;l[u]=tim;q[tim]=u;
for (int i=1;i<=16;i++)
if (fac[i]<=dep[u]) fa[u][i]=fa[fa[u][i-1]][i-1];
else break;
for (int i=Head[u];i;i=Next[i])
{
fa[ret[i]][0]=u;
dep[ret[i]]=dep[u]+1;
dfs(ret[i]);
}
r[u]=tim;
}
void build(int k,int l,int r)
{
if (l==r){mn[k]=val[q[l]];return;}
int mid=(l+r)>>1;
build(k<<1,l,mid);build(k<<1|1,mid+1,r);
mn[k]=min(mn[k<<1],mn[k<<1|1]);
}
void modify(int k,int l,int r,int x,int y)
{
if (l==r){mn[k]=y;return;}
int mid=(l+r)>>1;
if (x<=mid) modify(k<<1,l,mid,x,y);
else modify(k<<1|1,mid+1,r,x,y);
mn[k]=min(mn[k<<1],mn[k<<1|1]);
}
int query(int k,int l,int r,int x,int y)
{
if(x>y)return inf;
if (l==x&&r==y) return mn[k];
int mid=(l+r)>>1;
if (y<=mid) return query(k<<1,l,mid,x,y);
else if (x>mid) return query(k<<1|1,mid+1,r,x,y);
else return min(query(k<<1,l,mid,x,mid),query(k<<1|1,mid+1,r,mid+1,y));
}
int main()
{
fac[0]=1;for (int i=1;i<20;i++) fac[i]=fac[i-1]<<1;
n=read();Q=read();
for (int i=1;i<=n;i++)
{
int f=read();val[i]=read();
if (f) ins(f,i);
}
rt=1;
dfs(1);
build(1,1,n);
char ch[5];int x;
while (Q--)
{
scanf("%s",ch);x=read();
if (ch[0]=='V')
{
int val=read();
modify(1,1,n,l[x],val);
}
else if (ch[0]=='E') rt=x;
else
{
if (rt==x) printf("%d\n",mn[1]);
else if (l[x]<=l[rt]&&r[x]>=r[rt])
{
int y=rt,d=dep[y]-dep[x]-1;
for(int i=0;i<=16;i++)
if(fac[i]&d)y=fa[y][i];
printf("%d\n",min(query(1,1,n,1,l[y]-1),query(1,1,n,r[y]+1,n)));
}
else printf("%d\n",query(1,1,n,l[x],r[x]));
}
}
}