题意:有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;
}