小G有一个大树
來(lái)源:牛客網(wǎng)
:
題目描述
小G想要把自己家院子里的橘子樹(shù)搬到家門口(QAQ。。就當(dāng)小G是大力水手吧)
可是小G是個(gè)平衡性灰常灰常差的人,他想找到一個(gè)這個(gè)橘子樹(shù)的平衡點(diǎn)。 怎么描述這棵樹(shù)呢。。。就把它看成由一個(gè)個(gè)節(jié)點(diǎn)構(gòu)成的樹(shù)吧。結(jié)點(diǎn)數(shù)就
代表樹(shù)重。
輸入描述:
多組數(shù)據(jù)輸入輸出, 第一行包含一個(gè)整數(shù)n(3<=n<=1000)代表樹(shù)的結(jié)點(diǎn)的個(gè)數(shù) 以下n-1行描述(1-n)節(jié)點(diǎn)間的連接關(guān)系。 輸出描述:
輸出兩個(gè)個(gè)整數(shù) x,num 分別代表樹(shù)的平衡點(diǎn),和刪除平衡點(diǎn)后最大子樹(shù)的結(jié)點(diǎn)數(shù)(如果結(jié)點(diǎn)數(shù)相同輸出編號(hào)小的)。
示例1
輸入
復(fù)制
輸出
復(fù)制
題解:
樹(shù)型dp入門
狀態(tài)轉(zhuǎn)移方程:dp[i]=max(n-dp[i],max(dp[k])),k是i的兒子
dp[x]:表示以x為根的樹(shù)的節(jié)點(diǎn)數(shù)
在dfs過(guò)程中把狀態(tài)轉(zhuǎn)移方程帶進(jìn)去
代碼:
#include<bits/stdc++.h> using namespace std; const int maxn=1010; vector<int>edge[maxn]; int num,ans; int dp[maxn]; int n; void dfs(int u,int fa) {dp[u]=1;int maxx=0;for(int i=0;i<edge[u].size();i++){int v=edge[u][i];if(v==fa)continue;dfs(v,u);dp[u]+=dp[v];//v是u的子樹(shù) maxx=max(dp[v],maxx);}maxx=max(maxx,n-dp[u]);//判斷是這個(gè)子樹(shù)重,還是另一個(gè) // dp[i]=max(n-tot[i],max(tot[k])) 樹(shù)形dp if(maxx<num)//刪除平衡點(diǎn)后最大子樹(shù)的結(jié)點(diǎn)數(shù){num=maxx;ans=u; }else if(maxx==num){if(ans>u)ans=u;//如果結(jié)點(diǎn)數(shù)相同,我們要編號(hào)更小的 } } int main() {while(cin>>n){num=ans=0x7f;for(int i=1;i<=n;i++){edge[i].clear();dp[i]=0;}for(int i=1;i<n;i++){int u,v;cin>>u>>v;edge[u].push_back(v);edge[v].push_back(u);}dfs(1,0);cout<<ans<<" "<<num<<endl;} }總結(jié)
- 上一篇: 牛客网状压dp
- 下一篇: 【每日一题】8月10日题目精讲—排座椅