HDU 1102 Constructing Roads（Prim）

 

分别给出 i->j  的价格，给出 m 条路已经建好，求最小连接费用

const int N=100+5;
 
    int n,m,t;
    int i,j,k;
    int f[N][N];
    int minn[N];
    bool vis[N];

void Prim(int s)
{
    int ans=0;
    for(int i=1;i<=n;i++){
        vis[i]=0;
        minn[i]=f[s][i];
    }
    minn[s]=0;
    vis[s]=1;
    for(int i=1;i<=n-1;i++){
        int flag,dis=inf;
        for(j=1;j<=n;j++){
            if(!vis[j] && dis>minn[j]){
                dis=minn[j];
                flag=j;
            }
        }
        vis[flag]=1;
        ans+=dis;
        for(j=1;j<=n;j++){
            if(!vis[j] && minn[j]>f[flag][j])
                minn[j]=f[flag][j];
        }
    }
    pd(ans);
}
int main()
{
    //IOS;
    while(sd(n)==1){
        for(i=1;i<=n;i++){
            for(j=1;j<=n;j++) sd(f[i][j]);
        }
        sd(m);
        while(m--){
            int u,v;
            sdd(u,v);
            f[u][v]=f[v][u]=0;
        }
        Prim(1);
    }
    //PAUSE;
    return 0;
}