51Nod 1737 配对（树的重心）

http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1737

题意：

思路：

树的重心。

树的重心就是其所以子树的最大的子树结点数最少，删除这个点后最大连通块的结点数最小，也就说各个连通块尽量平衡。

这道题的话就是先求一个重心，然后求各个点到重心的距离之和。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<vector>
#include<queue>
#include<cmath>
#include<map>
using namespace std;

const int maxn=1e5+5;
const int INF=0x3f3f3f3f;

int n;

vector<pair<int,int> > edge[maxn];

int zx,sz;
int son[maxn],vis[maxn];
long long ww[maxn];   //各个点到重心的距离

void init()
{
    for(int i=1;i<=n;i++)
        edge[i].clear();
    memset(vis,0,sizeof(vis));
    memset(ww,0,sizeof(ww));
    sz=INF;
    zx=-1;
}

void dfs(int u)
{
    vis[u]=1;
    son[u]=0;
    int temp=0;
    for(int i=0;i<edge[u].size();i++)
    {
        int v=edge[u][i].first;
        if(!vis[v])
        {
            dfs(v);
            son[u]+=son[v]+1;    //找到该子树中的最大结点数
            temp=max(temp,son[v]+1);
        }
    }
    temp=max(temp,n-son[u]-1);    //删去重心后子树最大结点数和非重心的子树比较
    if(temp<sz)
    {
        zx=u;
        sz=temp;
    }
}

void sum(int u)
{
    vis[u]=1;
    for(int i=0;i<edge[u].size();i++)
    {
        int v=edge[u][i].first;
        if(!vis[v])
        {
            ww[v] =ww[u]+edge[u][i].second;
            sum(v);
        }
    }
}

int main()
{
    //freopen("D:\input.txt","r",stdin);
    while(~scanf("%d",&n))
    {
        init();
        int u,v,w;
        for(int i=1;i<=n-1;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            edge[u].push_back(make_pair(v,w));
            edge[v].push_back(make_pair(u,w));
        }

        dfs(1);
        memset(vis,0,sizeof(vis));
        sum(zx);
        long long ans=0;
        for(int i=1;i<=n;i++)
            ans+=ww[i];
        printf("%lld
",ans);
    }
}