Poj(2421)，Prim最小生成树

题目链接：http://poj.org/problem?id=2421

最小生成树的变形，有的村庄已经连接了，就直接把他们的权值赋为0，一样的做最小生成树，Prim算法。

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

#define INF 0x3f3f3f3f

int maps[][];
bool vis[];
int dis[];
int n;

int Prim()
{
    memset(vis,false,sizeof(vis));
    for(int i=;i<=n;i++)
        dis[i] = INF;
    int ans=;
    dis[] = ;
    for(int i=;i<=n;i++)
    {
        int tmp = INF,k=;
        for(int j=;j<=n;j++)
        {
            if(!vis[j]&&dis[j]<tmp)
            {
                tmp = dis[j];
                k=j;
            }
        }

        vis[k] = true;
        ans+=tmp;
        for(int i=;i<=n;i++)
        {
            if(!vis[i]&&dis[i]>maps[k][i])
                dis[i] = maps[k][i];
        }

    }
    return ans;
}

int main()
{

    scanf("%d",&n);
    for(int i=; i<=n; i++)
        for(int j=; j<=n; j++)
            scanf("%d",&maps[i][j]);
    int k;
    scanf("%d",&k);
    for(int i=; i<k; i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        maps[a][b]=maps[b][a] = ;
    }
    printf("%d\n",Prim());
    return ;
}