A - 畅通工程 (prim+ 链式前向星)

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。经过调查评估,得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序,计算出全省畅通需要的最低成本。 

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;
}

猜你喜欢

转载自blog.csdn.net/nucleare/article/details/81121817