树直径和树重心

mac2024-05-12  31

主要是留个板子

因为我似乎不会证明

树直径

丛树的任一点开始找一个离该节点最远的点,该点必为树直径的一端点,再从该点出发找最远点即为树直径

方法:dfs  时间O(2n)

 

 

 

 

树重心

以某个节点为树根,最大子树最小的节点

方法是树dp  时间O(n)

要注意的点都在代码注释里

#include<bits/stdc++.h> using namespace std; typedef long long ll; const ll INF=0x3f3f3f; const int N=5e5+7; int dp[N];///存该节点不走父边的所有节点和 vector<int>son[N];///vector邻接表存树 int mi;///最大子树最小值 int n,m; int a[N];///以下标为节点的树的最大子树节点数 void dfs(int x,int fa) { dp[x]=1; a[x]=0; for(int i=0;i<son[x].size();i++) { if(son[x][i]==fa) continue; dfs(son[x][i],x); dp[x]+=dp[son[x][i]]; a[x]=max(a[x],dp[son[x][i]]); } a[x]=max(a[x],n-dp[x]); if(a[x]<mi) mi=a[x]; } int main() { ios::sync_with_stdio(false); cin>>n; m=n-1; mi=n; while(m--) { int a1,a2; cin>>a1>>a2; son[a1].push_back(a2); son[a2].push_back(a1); } dfs(1,0); for(int i=1;i<=n;i++) if(a[i]==mi) cout<<i<<' ';cout<<endl; }

 

 

最新回复(0)