POJ 1258 Agri-Net(最小生成树Prim算法)

//
//  main.cpp
//  Richard
//
//  Created by 邵金杰 on 16/8/17.
//  Copyright © 2016年 邵金杰. All rights reserved.
//


#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;
struct edge{
    int v,w;
    edge(int v,int w): v(v),w(w) {}
    bool operator < (const edge p) const
    {
        return w>p.w;
    }
};
const int INF=100000000;
int vis[110],dist[110];
vector<vector<edge> > G(110);
int HeapPrime(int n)
{
    memset(vis,0,sizeof(vis));
    priority_queue<edge> pq;
    for(int i=0;i<n;i++)
        dist[i]=INF;
    pq.push(edge(0,0));
    int nCount=0;
    int mindist=0;
    while(nCount<n&&!pq.empty())
    {
        edge nd(0,0);
        do{
            nd=pq.top();pq.pop();
        }while(vis[nd.v]==1&&!pq.empty());
        if(vis[nd.v]==0)
        {
            vis[nd.v]=1;
            nCount++;
            mindist+=nd.w;
            for(int i=0;i<G[nd.v].size();i++)
            {
                int v=G[nd.v][i].v;
                if(vis[v]==0)
                {
                    int w=G[nd.v][i].w;
                    if(dist[v]>w){
                        dist[v]=w;
                        pq.push(edge(v,w));
                    }
                }
            }
        }
    }
    if(nCount<n) return -1;
    return mindist;
}
int main()
{
    int n,w;
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=0;i<n;i++)
            G[i].clear();
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                scanf("%d",&w);
                G[i].push_back(edge(j,w));
            }
        }
        printf("%d\n",HeapPrime(n));
    }
    return 0;
}