poj 2421 Constructing Roads mst(…

mst问题

对已经建好的路把路径更改为一，然后就可以上经典的prim算法

或者可以用库鲁思卡尔算法，建好的路径相当于把集合并在了一起，不过这样的话，克鲁斯卡尔结束的条件就要修改成当所有的元素都在同一个集合里面

#include <iostream>
#include<stdio.h>
#include<string.h>
int d[101][101];
using namespace std;

int min(int key[],int n)
{
    int ret=99999,tmp;
    for(int i=1;i<=n;i++)
    if(key[i]!=-1&&key[i]<ret)
    {
        ret=key[i];
        tmp=i;
    }
    return(tmp);
}

int prim(int n)
{
    int key[101];
    for(int i=1;i<=n;i++)
    key[i]=d[1][i];
    key[1]=-1;
    int ans=0;
    for(int k=2;k<=n;k++)
    {
        int u=min(key,n);
        ans+=key[u];
        key[u]=-1;
        for(int i=1;i<=n;i++)
        if(key[i]>d[u][i])
        key[i]=d[u][i];
    }
    return(ans);
}

int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    scanf("%d",&d[i][j]);
    int m;
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        int a,b;
        scanf("%d %d",&a,&b);
        d[a][b]=0;
        d[b][a]=0;
    }
    int ans=prim(n);
    printf("%d\n",ans);
    return 0;
}