Problem Description
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。经过调查评估,得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序,计算出全省畅通需要的最低成本。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 );随后的 N
行对应村庄间道路的成本,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间道路的成本(也是正整数)。为简单起见,村庄从1到M编号。当N为0时,全部输入结束,相应的结果不要输出。
行对应村庄间道路的成本,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间道路的成本(也是正整数)。为简单起见,村庄从1到M编号。当N为0时,全部输入结束,相应的结果不要输出。
Output
对每个测试用例,在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通,则输出“?”。
Sample Input
3 3 1 2 1 1 3 2 2 3 4 1 3 2 3 2 0 100
Sample Output
3 ?
/*
用优先队列优化
*/
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include <vector>
using namespace std;
#define maxn 108
bool vis[maxn];
int dis[maxn];
int n,m;
struct Edge
{
int v;
int w;
bool operator < (const Edge & a) const
{
return w > a.w;
}
}edge;
vector <Edge> ans[maxn];
int heapprime()
{
priority_queue <Edge> q;
int cost = 0;
edge.v = 1;
edge.w = 0;
q.push(edge);
int nown = 0,nowv;
while(nown < n && !q.empty())
{
L:
edge = q.top();
q.pop();
while(vis[edge.v] && !q.empty()) goto L;
if(!vis[edge.v])
{
nown++;
vis[edge.v] = 1;
cost += edge.w;
nowv = edge.v;
for(int i=0;i<ans[nowv].size();i++)
{
if(!vis[ans[nowv][i].v])
{
if(dis[ans[nowv][i].v] > ans[nowv][i].w)
{
dis[ans[nowv][i].v] = ans[nowv][i].w;
edge.v = ans[nowv][i].v;
edge.w = ans[nowv][i].w;
q.push(edge);
}
}
}
}
}
if(nown < n)return -1;
else return cost;
}
int main()
{
while(scanf("%d%d",&m,&n)==2 && m)
{
int u,v,w;
memset(vis,0,sizeof(vis));
memset(dis,0x3f,sizeof(dis));
for(int i=1;i<=n;i++)
{
ans[i].clear();
}
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&u,&v,&w);
edge.v = v;
edge.w = w;
ans[u].push_back(edge);
edge.v = u;
edge.w = w;
ans[v].push_back(edge);
}
int ans = heapprime();
if(ans != -1)
{
printf("%d\n",ans);
}
else printf("?\n");
}
return 0;
}
/*
不用优先队列优化
*/
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
#define maxn 108//村庄个数
#define inf 0x3f3f3f3f
int dis[maxn];
int W[maxn][maxn];//这个直接就可以看出两个村庄是不是相连的
bool vis[maxn];
inline int min(int a,int b)
{
return a>b?b:a;
}
int main()
{
int n,m;
while(scanf("%d%d",&m,&n)==2 && m)
{
memset(dis,0x3f,sizeof(dis));
memset(W,0,sizeof(W));
memset(vis,0,sizeof(vis));
int u,v,w;
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&u,&v,&w);
W[u][v] = W[v][u] = w;
}
dis[1] = 0;
int cost = 0;
bool flag = true;
for(int i=1;i<=n;i++)
{
int nowv,minlen=inf;
for(int j=1;j<=n;j++)
{
if(!vis[j])
{
if(dis[j] < minlen)
{
minlen = dis[j];
nowv = j;
}
}
}
if(minlen == inf)
{
flag = false;
break;
}
vis[nowv] = 1;
cost += dis[nowv];
for(int j=1;j<=n;j++)
{
if(!vis[j] && W[nowv][j])
{
dis[j] = min(dis[j],W[nowv][j]);
}
}
}
if(flag) printf("%d\n",cost);
else printf("?\n");
}
return 0;
}
Kruskal算法:
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#define maxn 108
int father[maxn];
int find(int x)
{
if(x == father[x])return x;
return father[x] = find(father[x]);
}
struct Edge
{
int from,to,key;
}edge[maxn*maxn];
bool cmp(Edge a,Edge b)
{
return a.key < b.key;
}
int main()
{
int n,m;
while(scanf("%d%d",&m,&n)==2 && m)
{
for(int i=1;i<=n;i++)
{
father[i]=i;
}
int u,v,w;
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&u,&v,&w);
edge[i].from = u;
edge[i].to = v;
edge[i].key = w;
}
if(m < n-1)
{
printf("?\n");
continue;
}
int z = 0,cost=0;
sort(edge+1,edge+m+1,cmp);
for(int i=1;i<=m;i++)
{
int uu=find(edge[i].from),vv=find(edge[i].to);
if(uu!=vv)
{
father[uu]=vv;
cost+=edge[i].key;
}
}
int zuzong=find(1);
bool flag=true;
for(int i=2;i<=n;i++)
{
if(find(i)!=zuzong)
{
flag=false;
break;
}
}
if(flag)printf("%d\n",cost);
else printf("?\n");
}
return 0;
}