hdu 4118(树形dp) 成都现场赛H题

题意：有一棵带权树每个节点上住着一个人。人后需要让所有人交换房子，但是每个结点只住一个人。问你所有人最长走多少路。

思路：这里有一个贪心就是每条边的权值乘以两边结点少的那一边的两倍。然后就是dfs求出每个结点子树上有多少点这样n-dp[v]就是另一边的点了。注意longlong和手动加栈。

代码如下：

/**************************************************
 * Author     : xiaohao Z
 * Blog     : http://www.cnblogs.com/shu-xiaohao/
 * Last modified : 2014-03-23 22:14
 * Filename     : hdu_chengdu1.cpp
 * Description     : 
 * ************************************************/

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <queue>
#include <stack>
#include <vector>
#include <set>
#include <map>
#define MP(a, b) make_pair(a, b)
#define PB(a) push_back(a)

using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<unsigned int,unsigned int> puu;
typedef pair<int, double> pid;
typedef pair<ll, int> pli;
typedef pair<int, ll> pil;

const int INF = 0x3f3f3f3f;
const double eps = 1E-6;
const int LEN = 100000+10;
vector<pii> Map[LEN];
int n, depth[LEN], dp[LEN], edge[LEN][3];

int dfs(int v, int f, int d){
    depth[v] = d;
    int ret = 1;
    for(int i=0; i<Map[v].size(); i++){
        if(Map[v][i].first!=f) ret+=dfs(Map[v][i].first, v, d+1);
    }
    dp[v] = ret;
    return ret;
}

int main()
{
//    freopen("in.txt", "r", stdin);

    ios::sync_with_stdio(false);
    int T, kase = 1, a, b, c;
    cin >> T;
    while(T--){
        for(int i=0; i<LEN; i++) Map[i].clear();
        cin >> n;
        for(int i=0; i<n-1; i++){
            cin >> a >> b >> c;
            a--;b--;
            edge[i][0] = a;edge[i][1] = b;edge[i][2] = c;
            Map[a].PB(MP(b, c));
            Map[b].PB(MP(a, c));
        }
        dfs(0, -1, 0);
        ll ans = 0;
        for(int i=0; i<n-1; i++){
            a = edge[i][0]; b = edge[i][1];
            c = (depth[a] > depth[b] ? dp[a] : dp[b]);
            ans += (min(c, n-c)*2*edge[i][2]);
        }
        cout << "Case #" << kase++ << ": ";
        cout << ans << endl;
    }
    return 0;
}