继续畅通工程
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Problem Description
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。现得到城镇道路统计表,表中列出了任意两城镇间修建道路的费用,以及该道路是否已经修通的状态。现请你编写程序,计算出全省畅通需要的最低成本。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 );随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态,每行给4个正整数,分别是两个村庄的编号(从1编号到N),此两村庄间道路的成本,以及修建状态:1表示已建,0表示未建。
当N为0时输入结束。
当N为0时输入结束。
Output
每个测试用例的输出占一行,输出全省畅通需要的最低成本。
Sample Input
3 1 2 1 0 1 3 2 0 2 3 4 0 3 1 2 1 0 1 3 2 0 2 3 4 1 3 1 2 1 0 1 3 2 1 2 3 4 1 0
Sample Output
3 1 0
刚开始还想直接将建好的边的点压入队列,但感觉比较麻烦。然后就想到如果边已存在,则将其权值设为0,然后就变成裸的最小生成树。
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
struct Edge {
int v,w;
Edge(int vv=0,int ww=INF):v(vv),w(ww) {}
bool operator < (const Edge& a) const {
return w>a.w;
}
}u;
int n,cnt,ans,vis[105],dis[105];
priority_queue <Edge> q;
vector<vector<Edge> > g;
void Prim() {
int i,j,v,w;
q.push(Edge(1,0));
memset(vis,false,sizeof(vis));
memset(dis,0x3f,sizeof(dis));
ans=cnt=0;
while(!q.empty()) {
u=q.top();
q.pop();
if(!vis[u.v]) {
vis[u.v]=true;
++cnt;
ans+=u.w;
for(i=0,j=g[u.v].size();i<j;++i)
if(!vis[v=g[u.v][i].v]&&dis[v]>(w=g[u.v][i].w)) {
dis[v]=w;
q.push(Edge(v,w));
}
}
}
}
int main() {
int i,s,e,cst,cond,t;
while(scanf("%d",&n),n) {
g.clear();
g.resize(n+1);
for(i=0,t=(n*(n-1))>>1;i<t;++i) {
scanf("%d%d%d%d",&s,&e,&cst,&cond);
g[s].push_back(Edge(e,cond==1?0:cst));//如果道路已建,则权值为0
g[e].push_back(Edge(s,cond==1?0:cst));
}
Prim();
printf("%d\n",ans);
}
return 0;
}