poj 1258 AgriNet

就是简单的最小生成树，本来用卡尔（Kruskal）算法写也没什么难的，可是偏偏要求用Prim写，搞得我很纠结，总算明白为什么白书里不说它了，因为写起来真的很麻烦，效率也不见得高多少。

#include <stdio.h>
#include <string.h>
#include <queue>
using namespace std;
int a[105][105];
bool vis[105];
typedef struct
{
    int a,b,w;
    void init(int x,int y,int z)
    {a = x;b= y; w = z;}
}Edge;
bool operator<(Edge m,Edge n)
{
    return m.w > n.w;
}
int main()
{
    int n,i,j,f,cnt,ans;
    Edge t;
    while(~scanf("%d",&n))
    {
        priority_queue <Edge> q;
        memset(vis,0,sizeof(vis));
        for(i = 1; i <= n; i++)
            for(j = 1; j <= n; j++)
                scanf("%d",&a[i][j]);
        cnt = 1;
        ans = 0;
        vis[1] = 1;
        for(j = 1; j <= n; j++)
            if(!vis[j])
            {
                t.init(1,j,a[1][j]);
                q.push(t);
            }
        while(cnt < n)
        {
            t = q.top();
            q.pop();
            if(!(vis[t.a] && vis[t.b]))
            {
                ans += t.w;
                cnt++;
            //    printf("%d\n",t.w);
                f = vis[t.a] ?t.b :t.a ;
                vis[f] = 1;
            for(j = 1; j <= n; j++)
                if(!vis[j])
                {
                    t.init(f,j,a[f][j]);
                    q.push(t);
                }
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}