hihocoder 1097 最小生成树一·Prim算法

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<set>
#include<vector>

using namespace std;
#define INF 0x3f3f3f3f
#define eps 1e-10
typedef long long ll;
const int maxn = 1e3+5;
const int mod = 1e9 + 7;
int gcd(int a, int b) {
    if (b == 0) return a;  return gcd(b, a % b);
}

int cost[maxn][maxn];
int mincost[maxn];
bool used[maxn];
int n;

int prim()
{
    mincost[1]=0;
    int res=0;
    while(true)
    {
        int v=-1;
        for(int u=1;u<=n;u++)
            if(!used[u] && (v==-1 || mincost[u]<mincost[v]))
                v=u;
        if(v==-1)
            break;
        used[v]=true;
        res+=mincost[v];
        for(int u=1;u<=n;u++)
            mincost[u]=min(mincost[u],cost[u][v]);
        
    }
    return res;
}
int main()
{

    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        mincost[i]=INF;
        used[i]=false;
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            int l;
            scanf("%d",&l);
            cost[i][j]=l;
        }
    cout<<prim()<<endl;
}