脑子清醒的时候再写题,否则再简单的题也是写不出来的!
代码:
#include<bits/stdc++.h>
using namespace std;
typedef struct
{
int u;
int v;
int w;
}dge;
dge e[10010];
int cmp(dge x,dge y)
{
return x.w<y.w;
}
int main()
{
int n,m,i,a,b,c,vest[10010],k,j,u1,v1,sn1,sn2,sum=0;
while(cin>>n>>m)
{
sum=0;
for(i=0;i<m;i++)
{
cin>>a>>b>>c;
e[i].u=a;
e[i].v=b;
e[i].w=c;
}
sort(e,e+m,cmp);
for(i=1;i<=n;i++)
vest[i]=i;
k=0;
for(j=0;j<m;j++)///真TM傻
{
u1=e[j].u;
v1=e[j].v;
sn1=vest[u1];
sn2=vest[v1];
if(sn1!=sn2)
{
sum=sum+e[j].w;
k++;
for(i=1;i<=n;i++)
if(vest[i]==sn2)
vest[i]=sn1;
}
}
if(k==n-1)
cout<<sum<<endl;
else
cout<<-1<<endl;
}
return 0;
}
没用prime,用的kruskal;一开始写错了是因为两个地方;一个是村庄数是从一开始的,不是从零开始的,是一到n而不是零到(n-1);其次是有可能题目给出的路径并不能把所有村庄都连起来,所以如果你用点做循环结束条件的话是肯定不对的(就是当所有村庄都被连接起来的时候循环结束,这是跟书学的;但书上是默认了一定存在最小生成树的,而我们现在是在写题。。。。。。);所以我们需要将所有的路径都遍历一遍才行!!!