1212 无向图最小生成树(prim)

N个点M条边的无向连通图，每条边有一个权值，求该图的最小生成树。
输入
第1行：2个数N,M中间用空格分隔，N为点的数量，M为边的数量。（2 <= N <= 1000, 1 <= M <= 50000)
第2 - M + 1行：每行3个数S E W，分别表示M条边的2个顶点及权值。(1 <= S, E <= N，1 <= W <= 10000)

输出
输出最小生成树的所有边的权值之和。

输入样例
9 14
1 2 4
2 3 8
3 4 7
4 5 9
5 6 10
6 7 2
7 8 1
8 9 7
2 8 11
3 9 2
7 9 6
3 6 4
4 6 14
1 8 8


输出样例
37

#include<bits/stdc++.h>
using namespace std;
int G[1010][1010],flag[1010],distan[1010];
int main()
{
  int n,m,s,e,w,ans=0,tip;
  cin>>n>>m;
  memset(flag,1,sizeof(flag));
  memset(G,0x3f3f3f3f,sizeof(G));
  for(int i=1;i<=m;i++)
  {
    cin>>s>>e>>w;
    G[s][e]=min(w,G[s][e]);
    G[e][s]=min(w,G[e][s]);
  } 
  for(int i=1;i<=n;i++)
   distan[i]=G[1][i];
  flag[1]=0;n;
  int N=n-1;
  while(N)
  {
   tip=0;
   for(int i=1;i<=n;i++)
    if(flag[i])
    {
     if(tip==0) tip=i;
   else if(distan[i]<distan[tip]) tip=i; 
  }
 ans+=distan[tip];
 flag[tip]=0;
 N--;
 for(int i=1;i<=n;i++)
  distan[i]=min(distan[i],G[tip][i]);
  }
  cout<<ans;
  return 0;
}