PTA 数据结构与算法题目集(中文) 7-10 公路村村通 (30分) 最小生成树(kruskal算法)


我的GIS/CS学习笔记:https://github.com/yunwei37/ZJU-CS-GIS-ClassNotes
<一个浙江大学本科生的计算机、地理信息科学知识库 >
还有不少数据结构和算法相关的笔记以及pta题解哦x


克鲁斯卡尔算法的基本思想是以边为主导地位,始终选择当前可用的最小边权的边(可以直接快排或者algorithm的sort)。每次选择边权最小的边链接两个端点是kruskal的规则,并实时判断两个点之间有没有间接联通。

PTA题目

现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。

输入格式:

输入数据包括城镇数目正整数N(≤1000)和候选道路数目M(≤3N);随后的M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见,城镇从1到N编号。

输出格式:

输出村村通需要的最低成本。如果输入数据不足以保证畅通,则输出−1,表示需要建设更多公路。

输入样例:

6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3

输出样例:

12

//Kruskal算法 
#include
#include

struct road{
	int v1;
	int v2;
	int cost;
} e[3004];//数组排序并选取最小值; 

int v[1001]={0};

int sort(const void *a,const void *b){
	return ((struct road*)a)->cost-((struct road*)b)->cost;
}

int root(int x){
	if(v[x]==0)	return x;
	else return v[x]=root(v[x]);
}

int main(){
	int n,m;
	scanf("%d%d",&n,&m);
	int i;
	for(i=0;i<m;++i)
		scanf("%d %d %d",&e[i].v1,&e[i].v2,&e[i].cost);		
	qsort(e,m,sizeof(struct road),sort);
	int sum=0;
	//并查集 
	for(i=0;i<m;i++){
		if(root(e[i].v1)!=root(e[i].v2)){
			v[root(e[i].v2)]=root(e[i].v1);// 路径压缩 
			sum+=e[i].cost;
		}
	}
	int flag=1;
	for(i=1;i<=n;i++){
		if(v[i]==0&&flag) flag=0;
		else if(v[i]==0) break;
	}
	if(i<=n)	printf("-1");
	else
		printf("%d",sum);
}

你可能感兴趣的:(笔记,数据结构与算法,数据结构,c++,算法)