uva1664 并查集

先对所有边按容量由大到小排序，然后遍历每条边。对于一条边AB，先找到他们各自的并查集，如果没有就初始化他们自己的并查集和并查集中元素的个数cnt，然后判断中心城市在A并查集或者B并查集那个好，即cap[A]+theEdge[i].c*cnt[B]和cap[B]+theEdge[i].c*cnt[A]那个大，然后将小的加入大的，更新cap和cnt。注意数据范围，cap使用long long。

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<string>
#include<map>
#include<set>
#include<algorithm>
#include<vector>
#include<queue>
#include<stack>
#include<sstream>
#define ll long long
#define OJ_DEBUG 0
#define READ_FILE 1
using namespace std;
const int NN_MAX = 200000+100;
//const int MM_MAX = NN_MAX*NN_MAX;
const int INF = 0x3fffffff;
struct Edge{
int l,r,c;
//Edge (int a1,int a2,int a3) :
//l(a1),r(a2),c(a3){}
Edge (){}
bool operator < (const Edge& rhs) const {return c>rhs.c;}
}theEdge[NN_MAX];
/**********************************************************/
int n;
int parent[NN_MAX],cnt[NN_MAX];
ll cap[NN_MAX];
/**********************************************************/
int min_2 (int x,int y) {return x<y?x:y;}
int max_2 (int x,int y) {return x>y?x:y;}
void swap (int& a, int& b){a^=b;b^=a;a^=b;}
int find (int i) {return i==parent[i]?i:parent[i]=find(parent[i]);}
/**********************************************************/
int main()
{
if (READ_FILE) freopen ("in.txt","r",stdin);
while (scanf ("%d",&n)!=EOF)
{
for (int i=0;i<n-1;i++)
scanf ("%d %d %d",&theEdge[i].l,&theEdge[i].r,&theEdge[i].c);
sort (theEdge,theEdge+n-1);
memset (parent,0,sizeof (parent));
memset (cap,0,sizeof (cap));
memset (cnt,0,sizeof (cnt));
for (int i=0;i<n-1;i++){
int x=theEdge[i].l, y=theEdge[i].r;
if (!parent[x]) {parent[x]=x;cnt[parent[x]]++;}
if (!parent[y]) {parent[y]=y;cnt[parent[y]]++;}
int px=find (x),py=find (y);
if (px!=py){
if (cap[px]+theEdge[i].c*cnt[py]>=cap[py]+theEdge[i].c*cnt[px]){
parent[py]=px;
cap[px]+=(ll)theEdge[i].c*(ll)cnt[py];
cnt[px]+=cnt[py];
}
else {
parent[px]=py;
cap[py]+=(ll) theEdge[i].c*(ll) cnt[px];
cnt[py]+=cnt[px];
}
}
}
printf ("%lld\n",cap[find(1)]);
}
return 0;
}