fzu 2028(dfs)

来源：互联网发布：万家三角梅淘宝店铺编辑：程序博客网时间：2024/06/11 01:14

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

0 0