题目
题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4118
题目来源:http://acm.hust.edu.cn/vjudge/contest/view.action?cid=105944#problem/J
简要题意:给定一棵树,每个点选择一个目标节点,要求每个点选一个,且不重复,求最大距离和。
题解
需要转换思路。模型比较经典。
dfs然后枚举每条边,去把边两边点数的最小值乘上边权加到答案里。
不会证明为啥要去这么做。
代码
#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <stack>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define sz(x) ((int)(x).size())
#define fi first
#define se second
using namespace std;
typedef long long LL;
typedef vector<int> VI;
typedef pair<int,int> PII;
// head
const int N = 1E5+5;
const int M = N<<1;
struct Edge {
int to, nxt, c;
Edge(int to, int nxt, int c) : to(to), nxt(nxt), c(c) {}
Edge() {}
};
int head[N];
Edge e[M];
void addEdge(int from, int to, int c, int cnt) {
e[cnt] = Edge(to, head[from], c);
head[from] = cnt;
}
void init(int n) {
for (int i = 0; i <= n; i++) head[i] = -1;
}
int cnt[N];
int t, n, cas = 1;
LL dfs(int x, int fa) {
cnt[x]++;
LL ans = 0;
for (int i = head[x]; ~i; i = e[i].nxt) {
int to = e[i].to;
if (to == fa) continue;
ans += dfs(to, x);
cnt[x] += cnt[to];
ans += (LL)e[i].c * min(n-cnt[to], cnt[to]);
}
return ans;
}
int main() {
scanf("%d", &t);
while (t--) {
scanf("%d", &n);
init(n);
memset(cnt, 0, sizeof cnt);
int u, v, c, ec = 0;
for (int i = 1; i < n; i++) {
scanf("%d%d%d", &u, &v, &c);
addEdge(u, v, c, ec++);
addEdge(v, u, c, ec++);
}
printf("Case #%d: %I64d\n", cas++, dfs(1, -1) * 2);
}
return 0;
}