poj 1861（prim）

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

题意：求最小生成树，并输出哪几个城市相连接，且输出一共有多少条边（一定是n-1条边），和最短边的长度

思路：直接prim，只不过在prim加点东西就可以，可以说是模板题，题目的案例是错的

答案应该是

1

3

1 2

1 3

2 4

#include <stdio.h>
#include <string.h>
#define inf 1000009
bool mark[1001];
int a[1001][1001],dis[1001],ans,n,m;
int b[1001];
int Min,Max;

int prim()
{
   Min = 0,ans = 0;
   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;
        if(tep>Min)
            Min = tep;
        mark[k]=false;
        for(int j=1;j<=n;j++)
            if(mark[j]&&dis[j]>a[k][j])
            {
                b[j] = k;    //用来记录哪两个城市相连接
                dis[j]=a[k][j];
            }
       }
   return 0;
}



int main()
{
    int x,y,z;
    while(~scanf("%d%d",&n,&m))
    {
        memset(a,inf,sizeof(a));
        memset(b,0,sizeof(b));
        memset(mark,true,sizeof(mark));
        while(m--)
        {
            scanf("%d%d%d",&x,&y,&z);
            a[x][y] = z;
            a[y][x] = z;
        }
        prim();
        printf("%d
%d
",Min,n-1);
        for(int i = 1;i<=n;i++)
        {
            if(b[i])
            {
                if(b[i]>i)
                    printf("%d %d
",i,b[i]);
                else printf("%d %d
",b[i],i);
            }
        }

    }
    return 0;
}