hdu 4118 Holiday's Accommodation 树上统计

题目

题目链接：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;
}