省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。经过调查评估,得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序,计算出全省畅通需要的最低成本。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 );随后的 N
行对应村庄间道路的成本,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间道路的成本(也是正整数)。为简单起见,村庄从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 ?
代码:
邻接矩阵+prim:
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int INF=0x3f3f3f3f;
int a[110][110],in[110],dis[110];
int main(void)
{
int n, m;
while(~scanf("%d",&m)&&m) {
scanf("%d",&n);
int u, v, w;
memset(a,0x3f,sizeof(a));
memset(in,0,sizeof(in));
memset(dis,0x3f,sizeof(dis));
for(int i = 1; i <= m; ++i) {
scanf("%d %d %d",&u, &v, &w);
a[u][v]=a[v][u]=w;
}
in[1]=1;
for(int i = 1; i <= n; ++i) {
if(!in[i] && dis[i] > a[i][1]) {
dis[i] = a[i][1];
}
}
int ed=0;
int sum=0;
for (int i = 1; i < n; ++i) {
int min_d=INF;
int min_i=-1;
for (int j = 1; j <= n; ++j) {
if(!in[j] && dis[j] < min_d) {
min_d = dis[j];
min_i=j;
}
}
if(min_i==-1)
break;
ed++;
sum += min_d;
in[min_i]=1;
for (int j = 1; j <= n; ++j) {
if(!in[j] && dis[j] > a[min_i][j]) {
dis[j] = a[min_i][j];
}
}
}
if(ed==n-1)
printf("%d\n",sum);
else
puts("?");
}
return 0;
}
代码:
链式前向星+prim
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int INF = 0x3f3f3f3f;
int head[10010], cnt=0;
int in[10010],dis[10010];
struct TA_
{
int to,next,w;
TA_() {}
TA_(int a,int b,int c):to(a),next(b),w(c){}
}ta[10010];
void addedge(int u,int v,int w) //链式前向星
{
ta[++cnt].to=v;
ta[cnt].next=head[u];
ta[cnt].w=w;
head[u]=cnt;
}
int main(void)
{
int n, m;
while(~scanf("%d",&m)&&m)
{
scanf("%d",&n);
int u, v, w;
cnt=0;
memset(head,-1,sizeof(head));
for (int i = 1; i <= m; ++i) {
scanf("%d %d %d",&u, &v, &w);
addedge(u, v, w);
addedge(v, u, w);
}
memset(dis,0x3f,sizeof(dis));
memset(in,0,sizeof(in));
in[1]=1;
for (int i = head[1]; ~i; i = ta[i].next) {
if(dis[ta[i].to] > ta[i].w)
dis[ta[i].to] = ta[i].w;
}
int ed=0;
int sum=0;
for(int i = 1; i < n; ++i) {
int min_d=INF;
int min_i=-1;
for(int j = 1; j <= n; ++j) {
if(!in[j] && dis[j] < min_d) {
min_d = dis[j];
min_i = j;
}
}
if(min_i==-1)
break;
sum += min_d;
ed++;
in[min_i] = 1; //标记
for(int j = head[min_i]; j; j=ta[j].next) { //越界?
if(j==-1) //yes
break;
if(!in[ta[j].to] && dis[ta[j].to] > ta[j].w) {
dis[ta[j].to] = ta[j].w;
}
}
}
if(ed==n-1)
printf("%d\n",sum);
else
puts("?");
}
return 0;
}