CDQZ Challenge 21

说在前面：“CDQZ”系列题目数据绝对良心（良苦用心233），提交网址如有需要请私信本蒟蒻。
1021:Challenge 21
查看 提交 统计 提问
总时间限制: 10000ms 单个测试点时间限制: 1000ms 内存限制: 256000kB
描述
一颗n(n<=2*10^5)个节点的树，初始时只有一号节点是关键点，支持两种m(m<=10^5)个操作，1、将一个点x设为关键点,2、询问一个点x与最近的关键点距离。
输入
第一行两个数n，m
第二行至第n行每行两个数x，y，表示树上x，y节点间有连边
第n+1行至第n+m+1行每行两个数opt，x，表示对x点做操作opt
输出
m行，对于每个操作输出一行表示答案
样例输入
5 5
1 2
1 3
3 4
3 5
2 5
2 3
1 3
2 3
2 4
样例输出
2
1
0
1
来源
mhy12345

题解：动态点分治！每次修改操作，更新与点x“连通”（就是指在x的vector里面的）的各分治重心的答案；每次询问操作，枚举所有与点x“连通”的分治重心，答案=min{分治重心的答案+x到该分治重心的距离}。

注意：寻找分治重心时，重心初始为0，所以0号点的mx[]（最小子树大小）要赋为INF！！！不然是找不出来滴orz。。。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=2e5+2,INF=0x3f3f3f3f;
struct NODE {
    int p,d;
};
int n,m,fa[maxn],dep[maxn],siz[maxn],q[maxn],mx[maxn],bg,ed,dis[maxn];
bool vis[maxn];
int head[maxn],edge=0;
struct EDGE {
    int v,nxt;
}e[maxn<<1];
vector<NODE> cs[maxn];
inline int read() {
    int x=0,f=1;char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x*f;
}
inline void adde(int u,int v) {
    e[edge].nxt=head[u],e[edge].v=v,head[u]=edge++;
    e[edge].nxt=head[v],e[edge].v=u,head[v]=edge++;
}
void vdcp(int u) {
    q[bg=ed=1]=u;
    fa[u]=0,siz[u]=0,mx[u]=0;
    while (bg<=ed) {
        int p=q[bg++];
        for (int i=head[p];~i;i=e[i].nxt) {
            int v=e[i].v;
            if (v==fa[p]||vis[v]) continue;
            q[++ed]=v,fa[v]=p,siz[v]=0,mx[v]=0;
        }
    }//build fa[],q[]
    for (register int i=ed;i;--i) {
        int p=q[i];
        ++siz[p];
        if (fa[p]) {
            siz[fa[p]]+=siz[p];
            mx[fa[p]]=max(mx[fa[p]],siz[p]);
        }
    }//calculate siz[],mx[]
    int c=0;mx[c]=INF;//INF!!!
    for (register int i=1;i<=ed;++i) {
        int p=q[i];
        mx[p]=max(mx[p],siz[u]-siz[p]);
        if (mx[p]<mx[c]) c=p;
    }//find the center of gravity
    q[bg=ed=1]=c;
    fa[c]=0,dep[c]=0;
    cs[c].push_back((NODE){c,dep[c]});
    while (bg<=ed) {
        int p=q[bg++];
        for (int i=head[p];~i;i=e[i].nxt) {
            int v=e[i].v;
            if (vis[v]||v==fa[p]) continue;
            q[++ed]=v,fa[v]=p,dep[v]=dep[p]+1,cs[v].push_back((NODE){c,dep[v]});
        }
    }
    vis[c]=true;
    for (int i=head[c];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (vis[v]) continue;//v could be c's father!!!
        vdcp(v);
    }
}
inline void modify(int pos) {
    int sz=cs[pos].size();
    for (int i=0;i<sz;++i) {
        int p=cs[pos][i].p,d=cs[pos][i].d;
        dis[p]=min(dis[p],d);
    }
}
inline int query(int pos) {
    int ans=INF,sz=cs[pos].size();
    for (int i=0;i<sz;++i) {
        int p=cs[pos][i].p,d=cs[pos][i].d;
        ans=min(ans,d+dis[p]);
    }
    return ans;
}
int main() {
//  freopen("Challenge 21.in","r",stdin);
    memset(head,-1,sizeof(head));
    n=read(),m=read();;
    for (register int i=1;i<n;++i) {
        int u=read(),v=read();
        adde(u,v);
    }
    vdcp(1);
    memset(dis,INF,sizeof(dis));
    modify(1);
    for (register int i=1;i<=m;++i) {
        int opt=read();
        if (opt^1) {
            int pos=read();
            printf("%d\n",query(pos));
        }
        else {
            int pos=read();
            modify(pos);
        }
    }
    return 0;
}