(题解来自北大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;
}