POJ 2421(prim)

http://poj.org/problem?id=2421

这个题和poj1258是一样的，只要在1258的基础上那么几行代码，就可以A，水。

题意：还是n连通问题，和1258不同的就是这个还有几条路在之前就已经连通了的，所以不需要再去连。

#include <stdio.h>
#include <string.h>


#define inf 100009

bool mark[1001];
int a[1001][1001],dis[1001],ans,n;

int prim()
{
   for(int i=1;i<=n;i++)
    dis[i]=inf;dis[1]=0;
    for(int i=1;i<=n;i++){
        int tep=inf;int k=0;
        for(int j=1;j<=n;j++){
            if(mark[j]&&dis[j]<tep)
            {
                tep=dis[j];
                k=j;
            }
        }
        if(tep==inf) return 0;
        ans+=tep;
        mark[k]=false;
        for(int j=1;j<=n;j++)
            if(mark[j]&&dis[j]>a[k][j])
                dis[j]=a[k][j];
       }
   return 0;
}

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        int x=0,c,b;
        memset(mark,true,sizeof(mark));
        ans=0;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                scanf("%d",&a[i][j]);
        scanf("%d",&x);
        while(x--)
        {
            scanf("%d%d",&b,&c);
            a[b][c]=0;
            a[c][b]=0;
        }
        prim();
        printf("%d
",ans);
    }
}