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;}