prim算法最小生成树（优化）

#include <iostream>
using namespace std;
const int maxn = 1000;
int f[maxn]={0};
int n,e;
int a[maxn][maxn]={0};
void prim(int v0) {
    int cost[maxn];
    for (int i=0;i<n;i++) cost[i]=1000000;
    for (int i=1;i<n;i++) if (a[v0][i]) cost[i]=a[v0][i];
    f[v0]=1;
    int pos;
    int ans=0;
    for (int k=1;k<n;k++) {
        int mini=1000000;
        for (int i=0;i<n;i++) {
            if (cost[i]<mini && !f[i]) {
                mini=cost[i];
                pos=i;
            }
        }
        f[pos]=1;ans += cost[pos];
        for (int i=0;i<n;i++) {
            if (!f[i] && a[pos][i] && a[pos][i]<cost[i]) {
                cost[i] = a[pos][i];
            }
        }
    }
}
int main () {
    cin >> n >> e;
    for (int i=0;i<e;i++) {
        int p,q;
        cin >> p >> q;
        cin >> a[p][q];
        a[q][p]=a[p][q];
    }
    prim(1);
}