HDU 4424 Conquer a New Region 最大生成树

来源：互联网发布：淘宝海外购物编辑：程序博客网时间：2024/06/16 11:30

给你一颗树每条边有一个权值选择一个点为中心定义S值为中心到其他n-1个点的路径上的最小边权求所有点S值的和

从大到小排序每次合并2棵树设为A集合 B集合设A集合的最大S值的和为sumA B集合为sumB

中心在A或者B现在加入A-B这条边使得2个集合连通因为A-B这条边的权值小于等于AB集合里面边的权值所以如果合并之后中心在A 那么sumA = sumA+B集合的点数*A-B这条边的权值 sumB = sumB+A集合的点数*A-B这条边的权值 2者取大

#include <cstdio>#include <cstring>#include <algorithm>#include <queue>#include <cmath>using namespace std;const int maxn = 200010;int n;int f[maxn], rank[maxn];__int64 sum[maxn];struct edge{int u, v, w;}e[maxn];int find(int x){if(x != f[x])return f[x] = find(f[x]);return f[x];}bool cmp(edge a, edge b){return a.w > b.w;}int main(){while(scanf("%d", &n) != EOF){for(int i = 0; i < n-1; i++)scanf("%d %d %d", &e[i].u, &e[i].v, &e[i].w);sort(e, e+n-1, cmp);for(int i = 0; i <= n; i++)f[i] = i, sum[i] = 0, rank[i] = 1;for(int i = 0; i < n-1; i++){int x = find(e[i].u);int y = find(e[i].v);if(x != y){if(sum[x]+(__int64)e[i].w*rank[y] > sum[y]+(__int64)e[i].w*rank[x]){f[y] = x;sum[x] = sum[x]+(__int64)e[i].w*rank[y];rank[x] += rank[y];}else{f[x] = y;sum[y] = sum[y]+(__int64)e[i].w*rank[x];rank[y] += rank[x];}}}int x = find(1);printf("%I64d\n", sum[x]);}return 0;}

0 0