NYoj38 布线问题 最小生成树Prim

它就是一个最小生成树的题目，也没有什么需要特殊处理的地方了- -

#include<stack>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define INF 200000
int a[501][501];
int dist[501];
int n,m,ans;
int b[501];
void prim()
{
    int vit[502];
    ans=0;
    memset(dist,0,sizeof(dist));
    memset(vit,0,sizeof(vit));
    for(int i=2;i<=n;i++)
        dist[i]=a[1][i];

    dist[1]=0;
    vit[1]=1;
    for(int i=2;i<=n;i++)
    {
        int MIN=INF;
        int flag;
        for(int j=1;j<=n;j++)
        {
            if(MIN>dist[j]&&!vit[j])
            {
                MIN=dist[j];
                flag=j;
            }
        }
        ans+=MIN;
        vit[flag]=1;
        for(int j=1;j<=n;j++)
        {
            if(!vit[j]&&dist[j]>a[flag][j])
                dist[j]=a[flag][j];
        }
    }
}
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        cin>>n>>m;
        memset(a,INF,sizeof(a));
        for(int i=1;i<=m;i++)
        {
           int x,y,z;
           cin>>x>>y>>z;
           a[x][y]=a[y][x]=z;
        }

        for(int i=1;i<=n;i++)
            cin>>b[i];
        sort(b+1,b+n+1);
        prim();
        cout<<ans+b[1]<<endl;
    }
}