公路村村通

输入样例:

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<iostream>
#include<vector>
#include<cmath>
#include<algorithm>
using namespace std;
int N,M;
//集数组Edge结构的定义
typedef struct{
	int begin;
	int end;
	int weight;
}Edge;
bool cmp(Edge a,Edge b)
{
	//return a.weight<b.weight;
    if(a.weight<b.weight) return true;
    else return false;
}
查找连线顶点的尾部下标
int Find(int *parent,int f)
{
	//cout<<f<<endl;
	while(parent[f]>0)
	f=parent[f];
	return f;
}
//Kruskal算法
void MiniSpanTree_Kruskal(Edge *edges)
{
	int i,n,m;
	int cnt=0,cost=0;
	int parent[1001];
	for(int i=1;i<=N;i++)
	parent[i]=0;
	for(i=1;i<=M;i++)
	{
		n=Find(parent,edges[i].begin);
		m=Find(parent,edges[i].end);
		//cout<<i<<" "<<edges[i].begin<<" "<<edges[i].end<<endl;
		//假如n与m不等，说明此边没有与现有生成树形成环路
		if(n!=m){
			cnt++;
			parent[n]=m;
			cost+=edges[i].weight;
		}
	}
	if(cnt==N-1) cout<<cost<<endl;
	else cout<<"-1"<<endl; 
}

int main()
{
    Edge edges[3001];
	cin>>N>>M;
	
	for(int i=1;i<=M;i++)
	{
		cin>>edges[i].begin>>edges[i].end>>edges[i].weight;
	}
	sort(edges+1,edges+M+1,cmp);
	MiniSpanTree_Kruskal(edges);
	return 0;
} 

Prime算法

#include<iostream>
#include<vector>
#include<cmath>
using namespace std;
int N,M;
void MiniSpanTree_Prim(vector<vector<int>> edges)
{
	int min,i,j,k,cost=0;
	int adjvex[N+1];
	int lowcost[N+1];
	//lowcost保存相关顶点间的权值,存入0表示已加入生成树
	lowcost[1]=0; 
	//adjvex保存相关顶点下标
	adjvex[1]=1;
	//将顶点间的权值存入 
	for(i=2;i<=N;i++)
	{
		lowcost[i]=edges[1][i];
		adjvex[i]=1;
	}
	//遍历除下标为1外的所有顶点 
	for(i=2;i<=N;i++)
	{
		min=INFINITY;
		j=2;k=1;
		//寻找lowcost最小权值和对应的下标 
		while(j<=N)
		{
			if(lowcost[j]!=0&&lowcost[j]<min)
			{
				min=lowcost[j];
				k=j;
			}
			j++;
		}
		if(k==1) break;
		//把权值加进来，并把lowcost下标为k处置0，表示已加入生成树 
		cost+=min;
		lowcost[k]=0;
		//更新在新顶点加进来后lowcost的权值，lowcost存入较小权值 
		for(j=1;j<=N;j++)
		{
			if(lowcost[j]!=0&&edges[k][j]<lowcost[j])
			{
				lowcost[j]=edges[k][j];
				adjvex[j]=k;
			}
		}
	} 
	if(i==N+1) cout<<cost<<endl;
	else cout<<"-1"<<endl;
}
int main()
{
	cin>>N>>M;
	vector<vector<int>> edges;
	edges.resize(N+1);
	for(int i=1;i<=N;i++)
	{
		edges[i].resize(N+1);
	}
	for(int i=1;i<=N;i++)
	{
		for(int j=1;j<N;j++)
		{
			edges[i][j]=INFINITY;
		}
	}
	for(int i=0;i<M;i++)
	{
		int va,vb,weight;
		cin>>va>>vb>>weight;
		edges[va][vb]=edges[vb][va]=weight;
	}
	MiniSpanTree_Prim(edges);
	return 0;
}