bzoj3306: 树

链接:http://www.lydsy.com/JudgeOnline/problem.php?id=3306

题意:中文题。

分析:因为题目要求查询子树中的最小值,很明显想到dfs序。但是还有一个操作:换根。这个问题看上去不好处理,但是其实自己画画图就能很好的将换根之后的情况进行处理了,假设当前询问x根为root,这是我们只要分类讨论:当x==root,return tr[1]。当lca(x,root)!=x直接return根为1时x子树中的最小值就可以了。当lca(x,root)-x时:这是其实我们画个图就能知道其实答案是整棵树中除了root到x路径上x的儿子的那颗子树剩下点的最小值。注意:有一点细节就是线段是的边界问题,当查询点的ou[i]==n时像我这样建树(区间是左闭右开的[l,r))的要特判一下,不然会死循环。

代码:

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=100100;
const int MAX=151;
const int mod=100000000;
const int MOD1=1000000007;
const int MOD2=1000000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=998244353;
const ll INF=10000000010;
typedef double db;
typedef unsigned long long ull;
int tot,w[N],u[N],v[N],pre[N];
void add(int x,int y) {
    v[tot]=y;pre[tot]=u[x];u[x]=tot++;
}
char s[4];
int k,a[N],in[N],ou[N],tr[4*N],de[N],f[N][20];
void dfs(int x,int y) {
    k++;in[x]=k;a[k]=w[x];
    f[x][0]=y;de[x]=de[y]+1;
    for (int i=1;i<20;i++)
    if (de[x]>1<<i) f[x][i]=f[f[x][i-1]][i-1];
    for (int i=u[x];i!=-1;i=pre[i]) dfs(v[i],x);
    ou[x]=k;
}
void buildtree(int x,int l,int r) {
    if (l+1==r) { tr[x]=a[l];return ; }
    int mid=(l+r)>>1;
    buildtree(x<<1,l,mid);
    buildtree((x<<1)+1,mid,r);
    tr[x]=min(tr[x<<1],tr[(x<<1)+1]);
}
void updata(int x,int l,int r,int a,int b) {
    if (l+1==r) { tr[x]=b;return ; }
    int mid=(l+r)>>1;
    if (a<mid) updata(x<<1,l,mid,a,b);
    else updata((x<<1)+1,mid,r,a,b);
    tr[x]=min(tr[x<<1],tr[(x<<1)+1]);
}
int getlca(int x,int y) {
    if (de[x]<de[y]) swap(x,y);
    for (int i=20;i>=0;i--)
    if (de[x]-(1<<i)>=de[y]) x=f[x][i];
    if (x==y) return x;
    for (int i=20;i>=0;i--)
    if (de[x]>(1<<i)&&f[x][i]!=f[y][i]) { x=f[x][i];y=f[y][i]; }
    return f[x][0];
}
int getmin(int x,int l,int r,int L,int R) {
    if (l==L&&r==R) return tr[x];
    int mid=(l+r)>>1;
    if (R<=mid) return getmin(x<<1,l,mid,L,R);
    else if (L>=mid) return getmin((x<<1)+1,mid,r,L,R);
        return min(getmin(x<<1,l,mid,L,mid),getmin((x<<1)+1,mid,r,mid,R));
}
int query(int x,int n,int root) {
    if (x==root) return tr[1];
    int i,lca=getlca(x,root);
    if (lca!=x) return getmin(1,1,n+1,in[x],ou[x]+1);
    for (i=20;i>=0;i--)
    if (de[root]-(1<<i)>de[x]) root=f[root][i];
    if (ou[root]==n) return getmin(1,1,n+1,1,in[root]);
    return min(getmin(1,1,n+1,1,in[root]),getmin(1,1,n+1,ou[root]+1,n+1));
}
int main()
{
    int i,n,m,x,y,root;
    scanf("%d%d", &n, &m);
    scanf("%d%d", &x, &w[1]);
    tot=0;memset(u,-1,sizeof(u));
    for (i=2;i<=n;i++) {
        scanf("%d%d", &x, &w[i]);add(x,i);
    }
    k=0;root=1;
    dfs(1,1);
    buildtree(1,1,n+1);
    while (m--) {
        scanf("%s", s);
        if (s[0]=='V') {
            scanf("%d%d", &x, &y);updata(1,1,n+1,in[x],y);
        }
        if (s[0]=='E') {
            scanf("%d", &x);root=x;
        }
        if (s[0]=='Q') {
            scanf("%d", &x);printf("%d\n", query(x,n,root));
        }
    }
    return 0;
}


你可能感兴趣的:(bzoj3306: 树)