POJ 1258生成树模板题

（题解来自北大2016 ACM/ICPC竞赛训练暑期课讲义）
题目链接：http://poj.org/problem?id=1258
题意：输入图的邻接矩阵，求最小生成树的总权值 (多组数据）
输入样例：
4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0
输出样例： 28

prioirty_queue实现 Prim + 堆

#include<iostream>
using namespace std;
#include<vector>
#include<cstdio>
#include<algorithm>
#include<queue>
#define maxn 1<<30
struct Edge
{
    int v; //边端点 另一端点已知
    int w; //边权值
    Edge(int v_=0,int w_=maxn):v(v_),w(w_){}
    bool operator<(const Edge&e)const
    {
        return w>e.w; //在队列中 边权值小的有限
    }
};
vector<vector <Edge> > G(110); //图的邻接表
int HeapPrim(const vector<vector <Edge> > &G,int n)
{
    Edge xDist(0,0);
    priority_queue<Edge>pq; //存放顶点及其到建生成树的距离
    vector<int>vDist(n);
    vector<int>vUsed(n);
    int nDoneNum=0;
    for (int i=0;i<n;i++)
    {
        vUsed[i]=0;
        vDist[i]=maxn;
    }
    nDoneNum=0;
    int nTotalW=0;
    pq.push(Edge(0,0));
    while (nDoneNum<n&&!pq.empty())
    {
        do
        { //每次从队列中拿出离生成树最近的点
            xDist=pq.top();
            pq.pop();
        }while (vUsed[xDist.v]==1&&!pq.empty());
        if (vUsed[xDist.v]==0)
        {
            nTotalW+=xDist.w;
            vUsed[xDist.v]=1;
            nDoneNum++;
            for (int i=0;i<G[xDist.v].size();i++) //对于新加入的点的邻点 更新他们
            {
                int k=G[xDist.v][i].v;
                if (vUsed[k]==0)
                {
                    int w=G[xDist.v][i].w;
                    if (vDist[k]>w)
                    {
                        vDist[k]=w;
                        pq.push(Edge(k,w));

                    }
                }
            }
        }
    }
    if (nDoneNum<n)
        return -1;
    return nTotalW; ///考察了所有边 且考察一条边时 可能执行pq.push(Edge(k,w)) 故复杂度Q(ElogV)
}
int main()
{
    int N;
    while (cin>>N)
    {
        for (int i=0;i<N;i++)
            G[i].clear();
        for (int i=0;i<N;i++)
        {
            for (int j=0;j<N;j++)
            {
                int w;
                cin>>w;
                G[i].push_back(Edge(j,w));
            }
        }
        cout<<HeapPrim(G,N)<<endl;
    }
    return 0;
}