题意
一棵个结点的树,令根节点的深度,其他节点的深度为,求以某个点为根节点这棵树的最小为多少?
思路
我们要枚举每一个结点为根节点的为多少,我们可以先求得以某片叶子为根节点的,然后再找寻相连的的关系,记录最小值。
以某一片叶子为根节点,DFS遍历整棵树,算其深度,算该节点的子树结点数量为。(无根树很多时候以叶子为结点可以避免讨论)
我们发现一个相连的的改变公式为
这里的即为以为根节点的与原先相反方向的子树结点个数,但是之所以不能用来表示是因为表示的是以一开始叶子为根节点的子树结点个数,由可知。
#include<bits/stdc++.h> #pragma GCC optimize(2) using namespace std; typedef long long ll; typedef pair<int,int>P; const double eps = 1e-8; const int NINF = 0xc0c0c0c0; const int INF = 0x3f3f3f3f; const ll mod = 1e9 + 7; const ll maxn = 1e6 + 5; const int N = 1e6 + 5; ll n,s[N],d[N],t,res; vector<int> G[N]; void dfs(int x,int fa){ s[x]=1; if(fa!=-1) d[x]=d[fa]+1; for(auto c:G[x]){ if(c==fa) continue; dfs(c,x); s[x]+=s[c]; } } void dp(int x,int fa,int y){ if(fa!=-1) res=min(res,y-s[x]+n-s[x]); for(auto c:G[x]){ if(c==fa) continue; dp(c,x,res); } } void solve(){ res=0; for(int i=1;i<=n;i++){ res+=d[i]; } dp(t,-1,res); } int main(){ ios::sync_with_stdio(false); cin.tie(0); cin>>n; for(int i=1;i<n;i++){ int x,y;cin>>x>>y; G[x].push_back(y); G[y].push_back(x); } for(int i=1;i<=n;i++){ if(G[i].size()==1) {t=i;dfs(i,-1);break;} } solve(); cout<<res<<'\n'; return 0; }