C语言最小生成树prim算法(USACO3.1)

/*
ID: hk945801
LANG: C++
TASK: agrinet
*/
#include<iostream>
#include<cstdio>
using namespace std;
int a[110][110],d[110],p[110];  
int main(){  
    int i,j,k,m,n,ans;
    //freopen("agrinet.in","r",stdin);
    //freopen("agrinet.out","w",stdout);
	cin>>n; 
    ans=0;  
    for(i=1;i<=n;i++)  
        for(j=1;j<=n;j++)  
            a[i][j]=0;    
    for(i=1;i<=n;i++){  
        d[i]=1000000000;  
        p[i]=0;  
    }  
    for(i=1;i<=n;i++)  
        for(j=1;j<=n;j++)  
            scanf("%d",&a[i][j]);  
    d[1]=0;  
    int min;  
    for(i=1;i<=n;i++){  
        min=1000000000;  
        for(j=1;j<=n;j++)  
            if(!p[j] && d[j]<min){  
                k=j;  
                min=d[j];  
            }  
        p[k]=1;  
        ans+=d[k];  
        for(j=1;j<=n;j++)  
            if(!p[j] && d[j]>a[j][k])  
                d[j]=a[j][k];  
    }  
    cout<<ans<<endl;   
    return 0;  
}