树的遍历

这是一篇不知所以然的方法总结……

因为昨天做了cf379的e，里面用到了O(n)求树的直径的方法，然后自己感觉这个和树分治里面寻找重心的方法很像，傻傻地来总结一下。

树有很多特殊性质，所以很多时候求树的某信息时可以用到O(n)的遍历算法？

大体套路是：

1. 选择某个点作为根节点，遍历树
2. 遍历树的同时记录每个节点的各个方向上需要统计的信息
3. 处理信息，得到所求结果，储存起来，供其祖先使用
4. 在最后一步检查答案能否更新

比如求树的直径代码：

int dp[MAXN];

void dfs(int u,int fa){

    int mx1=0,mx2=0;

    for(int i=head[u];~i;i=edge[i].nxt){

        int &v=edge[i].v;

        if(v==fa) continue;

        dfs(v,u);

        int val=dp[v]+1;

        mx2=max(mx2,val);

        if(mx1<mx2) swap(mx1,mx2);

    }

    dp[u]=mx1;

    ans=max(ans,mx1+mx2);

}

这个代码的行为就是：

1. 选择某个点作为根节点，遍历树
2. 遍历每个节点的同时不断更新这个节点，统计以这个节点为根的子链中的最长链和次长链
3. 确定以这个节点为根的最长链长度，储存，供其父节点使用
4. 在最后一步检查能否更新树的最长路（直径）

还有求树的重心的代码也是：

void dfs(int u,int f,int num){

    siz[u]=1;

    int K=0;

    for(int i=head[u];~i;i=edge[i].nxt){

        int &v=edge[i].v;

        if(used[v]||v==f) continue;

        dfs(v,u,num);

        K=max(K,siz[v]);

        siz[u]+=siz[v];

    }

    K=max(K,num-siz[u]);

    if(K<max_sub) max_sub=K,root=u;

}

这个代码的行为就是：

1. 选择某个点作为根节点，遍历树
2. 遍历每个节点的同时不断更新这个节点，统计以这个节点为根的子树中最重的子树
3. 确定以这个节点为根的子树的重量，储存，供其父节点使用
4. 在最后一步检查能否更新树的重心（重量分布最均匀的节点）

当然还有树链剖分时的求各子树重心的代码：

void dfs(int u,int fa,int d){

         dep[u]=d;

         pre[u]=fa;

         siz[u]=1;

         son[u]=0;

         int v;

         for(int i=head[u];~i;i=edge[i].next){

                  v=edge[i].v;

                  if(v==fa) continue;

                  dfs(v,u,d+1);

                  siz[u]+=siz[v];

                  if(siz[son[u]]<siz[v]) son[u]=v;

         }

}