树的简单问题总结

1. 树的重心

定义（参考 Oi Wiki）

对于树上的每一个点，计算其所有子树中最大的子树节点数，这个值最小的点就是这棵树的重心

重要性质

1. 以树的重心为根时，所有子树的大小都不超过整棵树大小的一半
2. 一棵树最多有两个重心
3. 树中所有点到某个点的距离和中，到重心的距离和是最小的；如果有两个重心，那么到它们的距离和一样
4. 在一棵树上添加或删除一个叶子，那么它的重心最多只移动一条边的距离
5. 把两棵树通过一条边相连得到一棵新的树，那么新的树的重心在连接原来两棵树的重心的路径上

求法（参考 博客）

1. 直接用定义求

inline void dfs(int now,int fa)
{
    siz[now]=val[now];
    dp[now]=0;
    for(int i=head[now];i;i=a[i].nxt)
    {
        int t=a[i].to;
        if(t==fa) continue;
        dfs(t,now);
        siz[now]+=siz[t];
        dp[now]=max(dp[now],siz[t]);
    }
    dp[now]=max(dp[now],n-dp[now]);
    if(dp[now]<dp[ans]) ans=now;
}

2. 用性质求（所有其他点到这个点的最短距离和）

inline void dfs1(int now,int fa,int deep)
{
    siz[now]=val[now];
    dep[now]=deep;
    int maxson=-1;//新 加 
    for(int i=head[now];i;i=a[i].nxt)
    {
        int t=a[i].to;
        if(t==fa) continue;
        dfs1(t,now,deep+a[i].val);
        siz[now]+=siz[t];
        f[now]+=f[t]+siz[t]*a[i].val;
        if(siz[t]>maxson) maxson=siz[t],son[now]=t;//新 加 
    }
}
inline void dfs2(int now,int fa)
{
    if(now^1) f[now]=f[fa]+siz[1]-2*siz[now];
    if(f[now]<sum) res=now,sum=f[now];
    if(son[now]) dfs2(t,now);//改 动 
}

动态维护

参考：https://www.cnblogs.com/qlky/p/5781081.html

2. 树的直径

定义

图中所有最短路径的最大值

求法

1. 两次DFS

const int N = 10009;
VI adj[N];
int d[N], c;
int n;
#define v (*it)
void dfs(int u) {
  ECH(it, adj[u]) if (!d[v]) {
    d[v] = d[u] + 1;
    if (d[v] > d[c]) c = v;
    dfs(v);
  }
}
#undef v
int main() {
  REP_C(i, RD(n) - 1) {
    int a, b;
    RD(a, b);
    --a, --b;
    adj[a].PB(b), adj[b].PB(a);
  }

  d[0] = 1, dfs(0);
  RST(d), d[c] = 1, dfs(c), OT(d[c] - 1);
}

2. 树形DP

#include <iostream>
#include <vector>
using namespace std;

const int N = int(1e4) + 9;
vector<int> adj[N];
int n, d;
int dfs(int u = 1, int p = -1) {
  int d1 = 0, d2 = 0;
  for (auto v : adj[u]) {
    if (v == p) continue;
    int d = dfs(v, u) + 1;
    if (d > d1)
      d2 = d1, d1 = d;
    else if (d > d2)
      d2 = d;
  }
  d = max(d, d1 + d2);
  return d1;
}
int main() {
  cin >> n;
  for (int i = 0; i < n - 1; ++i) {
    int a, b;
    cin >> a >> b;
    adj[a].push_back(b);
    adj[b].push_back(a);
  }
  dfs();
  cout << d << endl;
}