首页 > 代码库 > poj1655(树的重心)

poj1655(树的重心)

题意: 求树的重心;重心定义:以此节点为根的最大子树节点个数最小。


解法:dfs


代码:

/******************************************************
* @author:xiefubao
*******************************************************/
#pragma comment(linker, "/STACK:102400000,102400000")
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <queue>
#include <vector>
#include <algorithm>
#include <cmath>
#include <map>
#include <set>
#include <stack>
#include <string.h>
//freopen ("in.txt" , "r" , stdin);
using namespace std;

#define eps 1e-8
#define zero(_) (abs(_)<=eps)
const double pi=acos(-1.0);
typedef long long LL;
const int Max=20010;
const LL INF=0x3FFFFFFF;
vector<int> vec[Max];
int n;
int num[Max];
int ansnum;
int ans;
void dfs(int t,int be)
{
    int ma=0;
    int sum=0;
    for(int i=0;i<vec[t].size();i++)
    {
        if(vec[t][i]==be)
        continue;
        dfs(vec[t][i],t);
        ma=max(ma,num[vec[t][i]]);
        sum+=num[vec[t][i]];
    }
    ma=max(ma,n-sum-1);
    num[t]=sum+1;
    if(ansnum>ma||(ansnum==ma&&t<ans))
    {
        ansnum=ma;
        ans=t;
    }
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        scanf("%d",&n);
        ansnum=n+2;
        memset(num,0,sizeof num);
        for(int i=1; i<=n; i++)
            vec[i].clear();
        for(int i=0; i<n-1; i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            vec[a].push_back(b);
            vec[b].push_back(a);
        }
        dfs(1,0);
        cout<<ans<< " "<<ansnum<<endl;
    }
    return 0;
}

poj1655(树的重心)