#include <iostream>
#include <algorithm>
#include<stdio.h>
#include<math.h>
using namespace std;
struct point
{
int root;
}c[1001];
struct road
{
int s,e,w;
}r[5001];
bool cmp(road x,road y)
{
return x.w<y.w;
}
int main()
{
int N,M,i,rr,ww;
while(scanf("%d %d",&M,&N)!=EOF,M)
{
for(i=1;i<=N;i++)c[i].root=i;//初始化城镇的根
for(i=1;i<=M;i++)scanf("%d %d %d",&r[i].s,&r[i].e,&r[i].w);//录入边的权值信息
sort(r+1,r+M+1,cmp);
rr=0;ww=0;//初始化rr,ww
for(i=1;i<=M;i++)
{
if(rr==(N-1))break;
while(r[i].e!=c[r[i].e].root)
{
r[i].e=c[r[i].e].root;//找到根节点
}
while(r[i].s!=c[r[i].s].root)
{
r[i].s=c[r[i].s].root;//找到根节点
}
if(c[r[i].s].root<c[r[i].e].root){c[r[i].e].root=c[r[i].s].root;rr++;ww+=r[i].w;}
else if(c[r[i].s].root>c[r[i].e].root){c[r[i].s].root=c[r[i].e].root;rr++;ww+=r[i].w;}
}//最小生成树算法
if(rr==(N-1))printf("%d\n",ww);
else printf("?\n");
}
return 0;
}
总结
1.跟1N差不多,但要注意错误案例如下
#include <iostream>
#include <algorithm>
#include<stdio.h>
#include<math.h>
using namespace std;
struct point
{
int root;
}c[1001];
struct road
{
int s,e,w;
}r[5001];
bool cmp(road x,road y)
{
return x.w<y.w;
}
int main()
{
int N,M,i,rr,ww,tt;
while(scanf("%d %d",&M,&N)!=EOF,M)
{
for(i=1;i<=N;i++)c[i].root=i;//初始化城镇的根
for(i=1;i<=M;i++)scanf("%d %d %d",&r[i].s,&r[i].e,&r[i].w);//录入边的权值信息
sort(r+1,r+M+1,cmp);
rr=0;ww=0;//初始化rr,ww
for(i=1;i<=M;i++)
{
**************************************************************************************************************
if(rr==(N-1)){i=0;break;}//这里本意是把i=0作为执行成功的标志
**************************************************************************************************************
while(r[i].e!=c[r[i].e].root)
{
r[i].e=c[r[i].e].root;//找到根节点
}
while(r[i].s!=c[r[i].s].root)
{
r[i].s=c[r[i].s].root;//找到根节点
}
if(c[r[i].s].root<c[r[i].e].root){c[r[i].e].root=c[r[i].s].root;rr++;ww+=r[i].w;}
else if(c[r[i].s].root>c[r[i].e].root){c[r[i].s].root=c[r[i].e].root;rr++;ww+=r[i].w;}
}
**************************************************************************************************************
if(i)printf("?\n");
else printf("%d\n",ww);//但是实际上,当i==M执行最后一次循环时,rr++可能再执行一次,但是循环就结束了,不会再经过前面的判断是否成功的if语句,导致其实满足最小生成树,但是程序判断结果是不满足
**************************************************************************************************************
}
return 0;
}