题目描述
小G想要把自己家院子里的橘子树搬到家门口(QAQ。。就当小G是大力水手吧) 可是小G是个平衡性灰常灰常差的人,他想找到一个这个橘子树的平衡点。 怎么描述这棵树呢。。。就把它看成由一个个节点构成的树吧。结点数就 代表树重。
输入描述:
多组数据输入输出, 第一行包含一个整数n(3<=n<=1000)代表树的结点的个数 以下n-1行描述(1-n)节点间的连接关系。
输出描述:
输出两个个整数 x,num 分别代表树的平衡点,和删除平衡点后最大子树的结点数(如果结点数相同输出编号小的)。
示例1
输入
3 1 2 1 3
输出
1 1
一道经典的树上dp的问题也是一道求树的重心的模版题
这里给出树的重心的求法:
找到一个点 所有子树中最大的子树节点树最小
const int MAX=10010;
vector<int>G[MAX];
int dp[MAX];
int num=INF,ans=INF;
int n;
void dfs(int u,int fa){
dp[u]=1;
int sum=0;
for(int i=0;i<G[u].size();i++){
int v=G[u][i];
if(v==fa) continue;
dfs(v,u);
dp[u]+=dp[v];
sum=max(sum,dp[v]);
}
sum=max(sum,n-dp[u]);
if(sum<=ans){
ans=sum;
if(ans>sum) num=u;
else if(num>u) num=u;
}
}
int main(){
cin>>n;
for(int i=0;i<n-1;i++){
int u,v;
cin>>u>>v;
G[u].push_back(v);
G[v].push_back(u);
}
dfs(1,0);
cout<<num<<" "<<ans<<endl;
}