fzu 2028(dfs)

题意：有n个村子，然后给出了n-1条村子之间的路的长度，问从每个村子到其他村子的所走的路的总长度是多少。

题解：从给出的条件可以得到一个无根树，然后化为有根数，对于一条路走过的次数就是这条路父亲的结点数量 × 孩子的结点数量 × 2，然后再乘这条路的长度，就可以得到走这条路的总长度，递归的返回值代表当前节点有多少孩子节点，所以n - 返回值就是父亲结点数量。坑爹的是提交时如果用long long存总长度会wa。。。

#include <stdio.h>
#include <string.h>
#include <map>
#include <vector>
using namespace std;
const int N = 100005;
vector<int> g[N];
int t = 1, cases, n, vis[N];
__int64 sum;
map<pair<int, int>, int> path;

__int64 dfs(int u) {
	int d = g[u].size();
	__int64 temp = 1, h;
	for (int i = 0; i < d; i++) {
		int v = g[u][i];
		if (!vis[v]) {
			vis[v] = 1;
			h = dfs(v);
			sum += h * (n - h) * 2 * path[make_pair(u, v)];
			temp += h;
		}
	}
	return temp;
}

int main() {
	scanf("%d", &cases);
	while (cases--) {
		memset(vis, 0, sizeof(vis));
		path.clear();
		sum = 0;
		scanf("%d", &n);
		for (int i = 0; i < n; i++)
			g[i].clear();
		int a, b;
		__int64 l;
		for (int i = 0; i < n - 1; i++) {
			scanf("%d%d%I64d", &a, &b, &l);
			g[a].push_back(b);
			g[b].push_back(a);
			path[make_pair(a, b)] = path[make_pair(b, a)] = l;
		}
		vis[0] = 1;
		dfs(0);
		printf("Case %d: %I64d\n", t++, sum);
	}
	return 0;
}