【RQNOJ】【图论】宿命的P.S.S

题目描述

1.背景描述Background
P.S.S:“我来自哪里？”
WH:“你来自一个图。”
P.S.S:“我是谁？”
WH:“你是最小生成树。”
P.S.S:“我又要到哪里去？”
WH:“你要成为一个最小完全图（边权之和最小的完全图）。”
P.S.S:“为……为什么啊？”
WH:“这是你的宿命！因为你无聊！！！P.S.S！”
2.问题描述Description
最小生成树P.S.S在宿命的指引下找到了巫师Kismi。P.S.S希望Kismi能帮自己变成一个完全图。Kismi由于某些不可告人的原因，把这件事交给了你。
PS: 可以保证，这个最小生成树对于最后求出的完全图是唯一的。
7.时间限制Time Limitation
每个测试点1s。
8.数据规模：
Test1-Test10 n<60
Test11-Test25 n<20000
注释Hint: 
对于样例1，我们只需在1-3间加入一条边权为8的边即可得到最小完全图。其权和是4+7+8=19。对于样例2，我们在2-3间加入一条边权为2的边，在2-4和3-4间加入边权为3的边即可得到最小完全图，其权和是：1+1+2+2+3+3=12。
感谢吴豪牛对比赛的支持!!!

输入格式

输入的第一行是一个整数n，表示生成树的节点数。
接下来有n-1行，每行有三个正整数，依次表示每条边的端点编号和边权。
（顶点的边号在1-n之间，边权<maxint）

输出格式

一个整数ans，表示以该树为最小生成树的最小完全图的边权之和。

样例输入

样例输入1： 3 1 2 4 2 3 7 样例输入2： 4 1 2 1 1 3 1 1 4 2

样例输出

样例输出1: 19 样例输出2： 12

三维状态图像

---

同《测试数据》。

#include<stdio.h> #include<stdlib.h> struct ss { int a,b,w; } l[15001]; __int64 ans,n,t,i,j,k,ai,bi,wi,f[15001],tot[15001]; int cmp(const void*a,const void*b) { ss c=*(ss*)a,d=*(ss*)b; if (c.w>d.w) return 1; if (c.w<d.w) return -1; return 0; } int find(int x) { if (f[x]==x) return x; f[x]=find(f[x]); return f[x]; } void Union(int a,int b) { int fa=find(a),fb=find(b); if (tot[fa]>tot[fb]) { f[fb]=fa; tot[fa]+=tot[fb]; } else { f[fa]=fb; tot[fb]+=tot[fa]; } } int main() { scanf("%d",&n); ans=0; for (j=1;j<=n;++j) { f[j]=j; tot[j]=1; } for (j=0;j<n-1;++j) scanf("%d%d%d",&l[j].a,&l[j].b,&l[j].w); qsort(l,n-1,sizeof(ss),cmp); for (j=0;j<n-1;++j) { int fa=find(l[j].a),fb=find(l[j].b),tem=l[j].w; ans+=(tot[fa]*tot[fb]-1)*(tem+1)+tem; Union(fa,fb); } printf("%I64d/n",ans); return 0; }