题目:http://www.gdfzoj.com/oj/contest/276/problems/2
小杉坐在教室里,透过口袋一样的窗户看口袋一样的天空。
有很多云飘在那里,看起来很漂亮,小杉想摘下那样美的几朵云,做成棉花糖。
现在给你云朵的个数N,再给你M个关系,表示哪些云朵可以连在一起。小杉要把所有云朵连成K个棉花糖,一个棉花糖最少要用掉一朵云,小杉想知道他怎么连,花费的代价最小。
解释一下题目:(一开始连题目都没看懂)
首先由n朵云,即n个集合,要你通过连接云来合并集合最终得到k个集合,求其代价和(连接云有代价)最小值
这么一说,合并次数不就是n-k次咯?!
然后对于每一次合并,不就尽可能取最小的边?但又不能连“无用”边
那不就是 最小生成树 kruskal吗?
打一打顺便复习kruskal。。。开心~(原来kruskal是并查集呀,我这才知道)
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
struct node
{
int u,v,w;
};
const int maxS=10000;
node edge[maxS+5];
int n,m,k;
int a[maxS+5];
bool cmp(node x,node y)
{
if (x.w==y.w)
return x.u<y.u;
else
return x.w<y.w;
}
int find(int x)
{
int fu,x1;
if (a[x]==x)
return x;
else
fu=x;
while (a[fu]!=fu)
fu=a[fu];
while (a[x]!=fu)
{
x1=a[x];
a[x]=fu;
x=x1;
}
return fu;
}
void kruskal()
{
int i=1,tot=0,ans=0,fx,fy;
while (tot<n-k)
{
fx=find(edge[i].u); fy=find(edge[i].v);
if (fx!=fy)
{
a[fx]=fy;
ans+=edge[i].w; tot++;
}
i++;
if (i>m)
break;
}
if (tot<n-k)
printf("No Answer\n");
else
printf("%d\n",ans);
}
int main()
{
int i;
freopen("a.txt","r",stdin);
scanf("%d%d%d",&n,&m,&k);
for (i=1;i<=m;i++)
scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
sort(edge+1,edge+m+1,cmp);
for (i=1;i<=n;i++)
a[i]=i;
kruskal();
return 0;
}