首页 > 代码库 > XDOJ_1157_并查集
XDOJ_1157_并查集
http://acm.xidian.edu.cn/problem.php?id=1157
#include<iostream> #include<cstring> #include<cstdio> #include<algorithm> using namespace std; int point[100005]; struct line { int v,u,w; }l[1000005]; int n,pre[100005]; bool cmp(line x,line y) { return x.w < y.w; } int findd(int x) { int root = x; while(root != pre[root]) root = pre[root]; int temp; while(x != pre[x]) { temp = pre[x]; pre[x] = root; x = temp; } return root; } int main() { while(~scanf("%d",&n)) { long long ans = 0; for(int i = 1;i < n;i++) scanf("%d%d%d",&l[i].u,&l[i].v,&l[i].w); sort(l+1,l+n,cmp); for(int i = 1;i <= n;i++) { pre[i] = i; point[i] = 1; } for(int i = 1;i < n;i++) { int x = findd(l[i].u),y = findd(l[i].v); ans += (long long)point[x]*point[y]*l[i].w; pre[x] = y; point[y] += point[x]; } for(int i = 1;i <= n;i++) { pre[i] = i; point[i] = 1; } for(int i = n-1;i >= 1;i--) { int x = findd(l[i].u),y = findd(l[i].v); ans -= (long long)point[x]*point[y]*l[i].w; pre[x] = y; point[y] += point[x]; } printf("%lld\n",ans); } return 0; }
XDOJ_1157_并查集
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。