1O畅通工程-CSDN博客

本文链接：https://blog.csdn.net/DYNAZ/article/details/122666858

#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差不多，但要注意错误案例如下

**************************************************************************************************************
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;
}