prim算法求最小生成树

每次利用已经加入生成树的点集寻找未加入的点集的最小边，然后添加该点，并把minw[u]设置为最小生成树到u的最小距离；

证明算法正确性；

如果存在x1 x2均能联通到达生成树,其中若minw[x2]>minw[x1]

此时选x2加入生成树一定没有选X1优，正确性可以保证；

#include<bits/stdc++.h>
using namespace std;
const int N=5e3+5;
int n,m;
int dis[N][N];
int f[N],minw[N];

int main()
{
    memset(minw,0x3f,sizeof(minw));
    memset(dis,0x3f,sizeof(dis));
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        dis[u][v]=min(dis[u][v],w);
        dis[v][u]=min(dis[v][u],w);
    }
    minw[1]=0;
    for(int i=1;i<=n;i++)
    {
        int k=0;
        for(int j=1;j<=n;j++)
        {
            if(!f[j]&&minw[j]<minw[k])
            {
                k=j;
            }
        }
        f[k]=1;
        for(int j=1;j<=n;j++)
        {
            if(!f[j]&&(dis[k][j]<minw[j]||dis[j][k]<minw[j]))
            {
                minw[j]=dis[k][j];
            }    
        }
    }
    int ans=0,ok=0;
    for(int i=1;i<=n;i++)
    {
        if(minw[i]==0x3f3f3f3f)ok=1;
        else ans+=minw[i];
    }
    if(ok)printf("orz");
    else printf("%d",ans); 
    return 0;
}