公路村村通 prim算法

#include <iostream>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=1001;
bool visited[maxn]={false};
int dis[maxn];//表示i点到已被访问过点的最短距离
int graph[maxn][maxn];//存储图中节点的边权
int main()
{
      int n,m,cnt=0,sum=0;
      cin>>n>>m;
      for(int i=1;i<=n;++i) //初始化graph
      {
         for(int j=1;j<=n;++j)
         {
            if(i==j)
            {
                   graph[i][j]=0;
            }
            else
              graph[i][j]=INF;
         }
      }
      for(int i=0;i<m;++i)
      {
         int a,b,c;
         cin>>a>>b>>c;
         graph[a][b]=graph[b][a]=c;
      }
      //初始化dis
      for(int i=2;i<=n;++i)
      {
          if(graph[1][i]!=INF)
              dis[i]=graph[1][i];
          else
              dis[i]=INF;
      }
      visited[1]=true;
      int i;
      for(i=2;i<=n;++i)//将剩下的n-1个点也要加入s集合
      {
           int MIN=INF,u=-1;
           for(int j=1;j<=n;++j)
           {
               if(visited[j]==false&&dis[j]<MIN)
               {
                   MIN=dis[j];
                   u=j;
               }

           }
           if(u==-1)
              break;
           sum+=dis[u];
           visited[u]=true;
           for(int j=1;j<=n;++j)
           {
                if(visited[j]==false&&dis[j]>graph[u][j])
                {
                      dis[j]=graph[u][j];

                }

           }
      }
      if(i==n+1)
      {
        cout<<sum;
      }
      else
       cout<<-1;




    return 0;
}