数据结构专题系列 基础篇-8 最小生成树

数据结构专题系列 基础篇-8 最小生成树

--------------------------------------------------------------------------------

题目传送门

洛谷最小生成树 普及-组
洛谷最小生成树 普及/提高-组
--------------------------------------------------------------------------------
更多详见>>

OJ题解系列 目录导航帖
--------------------------------------------------------------------------------

这里是数据结构专题系列 基础篇-8 最小生成树
最小生成树作为树的一个分支,在简化图的算法中有着很好的应用,能够求解出连通图的最小边权和,同样,最小生成树会有许多新的变式,在基础部分我们只介绍最大生成树这一个扩展应用
基础部分的题目都是最小生成树的模板题(很多题目代码都基本一致,只是读入输出略有差别),第一道[模板]题,我们采用Prim算法加以解决,而剩余的其他题目我们则采用Kruskal算法加以解决,两个算法的复杂度分别为 O ( V l o g V + E ) , O ( ( V + E ) l o g V ) O(VlogV+E),O((V+E)logV) O(VlogV+E),O((V+E)logV)

接下来就是题解部分了,每道算法题都标注有对应的算法标签,对于那些易错、较难或是测试点比较特殊的题目会着重标注,本章推荐的题目有:

P2916 [USACO08NOV]Cheering up the Cow G树,二叉树 + 最小生成树(变式)
P4826 [USACO15FEB]Superbull S树,二叉树 + 最小生成树(最大生成树)
其余题目模板题(适合练习)

--------------------------------------------------------------------------------

难度: 普及-

P3366 【模板】最小生成树

算法标签: 树,二叉树 + 最小生成树
注意点: 模板题,这里用Prim算法实现

#include<bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f3f;
int g[5005][5005];
int vis[5005];
int d[5005];
long long ans = 0;

int N,M;
void Prim(int v0){
	int i,j,k,minn;
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=N;i++){
		d[i]=INF;
	}
	d[v0]=0;
	ans = 0;
	 
	for(int i=1;i<=N;i++){
		minn = INF;
		for(int j=1;j<=N;j++){
			if(vis[j]==0 && minn>d[j]){
				minn = d[j];
				k=j;
			}
		}
		vis[k]=1;
		ans+=d[k];
		
		for(int j=1;j<=N;j++){
			if(vis[j]==0 && d[j]>g[k][j]){
				d[j]=g[k][j];
			}
		}
		
	}
}
int main(){
	cin>> N >>M;
	for(int i=1;i<=N;i++){
		for(int j=1;j<=N;j++){
			g[i][j]=INF;
		}
	}
	for(int i=0;i<M;i++){
		int x,y,z;
		cin >>x >>y >> z;
		g[x][y]=min(g[x][y],z);
		g[y][x]=min(g[y][x],z);
	}
	Prim(1);
	cout << ans <<endl;
	
	return 0;
}

--------------------------------------------------------------------------------

难度: 普及/提高-

P2916 [USACO08NOV]Cheering up the Cow G

算法标签: 树,二叉树 + 最小生成树
注意点: 一道好题!考察了最小生成树的变形结构(本题采用kruskal算法):
由题意知,每条边和每个顶点至少被访问两次,因此最小生成树的边权=2个点权+边权*2,在读入数据的时候,就要对边权进行更新,此外,我们还需要选定一个起始顶点,这个点如果是两端的端点,那么不需要再计算一次(出发,返回,一共2次),但如果是内部顶点,还需计算一次(出发+回溯+返回,一共3次)

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e4+5;
int a[maxn];
struct node{
	int u,v;
	int dis;
};
node road[10*maxn];
bool cmp(node r1,node r2){
	return r1.dis < r2.dis;
}
int father[maxn];
int findfather(int x){
	if(x == father[x]){
		return x;
	}
	int F = findfather(father[x]);
	father[x] = F;
	return F;
}
void Union(int x,int y){
	int fx = findfather(x);
	int fy = findfather(y);
	if(fx!=fy){
		father[fx] = fy;
	}
}

int main(){
	int N,P;
	scanf("%d%d",&N,&P);
	int minn = INT_MAX;
	for(int i=1;i<=N;i++){
		scanf("%d",&a[i]);
		minn = min(minn,a[i]);
	}
	for(int i=0;i<P;i++){
		scanf("%d%d%d",&road[i].u,&road[i].v,&road[i].dis);
		road[i].dis *= 2;
		road[i].dis += a[road[i].u];
		road[i].dis += a[road[i].v];
	}
	sort(road,road+P,cmp);
	for(int i=1;i<=N;i++){
		father[i] = i;
	}
	int cnt = 0;
	int ans = 0;
	for(int i=0;i<P && cnt<N-1;i++){
		int u = road[i].u;
		int v = road[i].v;
		if(findfather(u)!=findfather(v)){
			Union(u,v);
			cnt++;
			ans += road[i].dis;
		}
	}
	cout << ans+minn << endl;
	return 0;
} 

P2330 [SCOI2005]繁忙的都市

算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
int n,m,s,t;
int prt[maxn];
typedef struct xiaoming{
	int x;
	int y;
	int z;
}xiaoming;
xiaoming a[maxn];

bool cmp(const xiaoming &x1,const xiaoming &x2){
	return x1.z<x2.z;
}
int Getfather(int x){
	if(prt[x]==x){
		return x;
	}
	prt[x] = Getfather(prt[x]);
	return prt[x];
}
void Kruskal(){
	int f1,f2,k;
	k=0;
	for(int i=1;i<=n;i++){
		prt[i]=i;
	}
	for(int i=1;i<=m;i++){
		f1 = Getfather(a[i].x);	
		f2 = Getfather(a[i].y);
		if(f1!=f2){
			prt[f1] = f2;
			k++;
			if(k==n-1){
				cout << a[i].z << endl;
				break;
			}
		}
	}
	
}
int main(){
	cin >> n >> m ;
	for(int i=1;i<=m;i++){
		cin >> a[i].x >> a[i].y >> a[i].z;
	}
	cout << n-1 << " ";
	sort(a+1,a+m+1,cmp);
	Kruskal();
	return 0;
}

P1396 营救

算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e4*2+5;
int n,m,s,t;
int prt[maxn];
typedef struct xiaoming{
	int x;
	int y;
	int z;
}xiaoming;
xiaoming a[maxn];
bool cmp(const xiaoming &x1,const xiaoming &x2){
	return x1.z<x2.z;
}
int Getfather(int x){
	if(prt[x]==x){
		return x;
	}
	prt[x] = Getfather(prt[x]);
	return prt[x];
}
void Kruskal(){
	int f1,f2,k;
	k=0;
	for(int i=1;i<=n;i++){
		prt[i]=i;
	}
	for(int i=1;i<=m;i++){
		f1 = Getfather(a[i].x);	
		f2 = Getfather(a[i].y);
		if(f1!=f2){
			prt[f1] = f2;
			k++;
			if(Getfather(s)==Getfather(t)){
				cout << a[i].z << endl;
				break;
			}
		}
	}
	
}
int main(){
	cin >> n >> m >> s >> t;
	for(int i=1;i<=m;i++){
		cin >> a[i].x >> a[i].y >> a[i].z;
	}
	sort(a+1,a+m+1,cmp);
	Kruskal();
	return 0;
}

P1546 [USACO3.1]最短网络 Agri-Net

算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
int n,m=0;
long long ans = 0;
int t[105][105];
int prt[maxn];
typedef struct xiaoming{
	int x;
	int y;
	int z;
}xiaoming;
xiaoming a[maxn];

bool cmp(const xiaoming &x1,const xiaoming &x2){
	return x1.z<x2.z;
}
int Getfather(int x){
	if(prt[x]==x){
		return x;
	}
	prt[x] = Getfather(prt[x]);
	return prt[x];
}
void Kruskal(){
	int f1,f2,k;
	k=0;
	for(int i=1;i<=n;i++){
		prt[i]=i;
	}
	for(int i=1;i<=m;i++){
		f1 = Getfather(a[i].x);	
		f2 = Getfather(a[i].y);
		if(f1!=f2){
			ans = ans +a[i].z;
			prt[f1] = f2;
			k++;
			if(k==n-1){
				cout << ans << endl;
				break;
			}
		}
	}
	
}
int main(){
	cin >> n;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cin >> t[i][j];
		}
	}
	for(int i=1;i<=n-1;i++){
		for(int j=i+1;j<=n;j++){
			++m;
			a[m].x=i;
			a[m].y=j;
			a[m].z=t[i][j];
		}
	}
	
	sort(a+1,a+m+1,cmp);
	Kruskal();
	return 0;
}

P2872 [USACO07DEC]Building Roads S

算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+1e3+5;
int prt[maxn];
double ans = 0.0;
typedef struct farm{
	int x;
	int y;
	double z;
}farm;
farm a[maxn];
int N,M;
int sum = 1;
bool cmp(const farm &x1,const farm &y1){
	return x1.z<y1.z;
} 
typedef struct city{
	int x;
	int y;
}city;
city c[maxn];

double dis(int x,int y){
	return (double)sqrt((double)(c[x].x-c[y].x)*(c[x].x-c[y].x)+(double)(c[x].y-c[y].y)*(c[x].y-c[y].y));
}

int Getfather(int x){
	if(prt[x]==x){
		return x;
	}
	prt[x] = Getfather(prt[x]);
	return prt[x];
}
void Kruskal(){
	int f1,f2,k;
	k=0;
	for(int i=1;i<=N;i++){
		prt[i]=i;
	}
	for(int i=1;i<sum;i++){
		f1 = Getfather(a[i].x);
		f2 = Getfather(a[i].y);
		if(f1!=f2){
			ans =  ans + a[i].z;
			prt[f1] = f2;
			k++;
			if(k==N-1){
				break;
			}
		}
	}
}
int main(){
	cin >> N >> M;
	for(int i=1;i<=N;i++){
		cin>> c[i].x >> c[i].y;
	}

	for(int i=1;i<N;i++){
		for(int j=i+1;j<=N;j++){
			a[sum].x = i;
			a[sum].y = j;
			a[sum++].z = dis(i,j);
		}
	}
	
	for(int i=sum;i<sum+M;i++){
		cin >> a[i].x >> a[i].y;
		a[i].z = 0;
	}
	
	sum = sum + M;
	sort(a+1,a+sum,cmp);
	Kruskal();
	printf("%.2lf\n",ans);
	
	return 0;
}

P1194 买礼物

算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal + 最优选取
注意题面中的这句话:特别的,如果 K I , J = 0 K_{I,J}=0 KI,J=0,那么表示这两样东西之间不会导致优惠,不要理解成可通过0元获取!

#include<bits/stdc++.h>
using namespace std;
const int maxn = 5e5+5;
int a[505][505];
int father[505];
int cnt = 0;
struct node{
	int x;
	int y;
	int dis;
};
node t[maxn];
bool cmp(node t1,node t2){
	return t1.dis < t2.dis;
}
int findfather(int x){
	if(x == father[x]){
		return x;
	}
	int F = findfather(father[x]);
	father[x] = F;
	return F;
}
void Union(int x,int y){
	int fx = findfather(x);
	int fy = findfather(y);
	if(fx!=fy){
		father[fx] = fy;
	}
}
int main(){
	int A,B;
	cin >> A >> B;
	for(int i=1;i<=B;i++){
		father[i] = i;
	}
	for(int i=1;i<=B;i++){
		for(int j=1;j<=B;j++){
			scanf("%d",&a[i][j]); 
		}
	}	
	for(int i=1;i<=B-1;i++){
		for(int j=i+1;j<=B;j++){
			if(a[i][j]){
				t[cnt].x = i;
				t[cnt].y = j;
				t[cnt].dis = a[i][j];
				cnt++;	
			}
				t[cnt].x = i;
				t[cnt].y = j;
				t[cnt].dis = A;
				cnt++;
		}
	}
	sort(t,t+cnt,cmp);
	int ans = 0;
	int count = 0;
	for(int i=0;i<cnt && count<B-1;i++){
		int u = t[i].x;
		int v = t[i].y;
		if(findfather(u)!=findfather(v)){
			Union(u,v);
			count++;
			ans += t[i].dis;
		}
	}
	cout << ans + A << endl;
	return 0;
} 

P1547 [USACO05MAR]Out of Hay S

算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
int n,m,s,t;
int prt[maxn];
typedef struct xiaoming{
	int x;
	int y;
	int z;
}xiaoming;
xiaoming a[maxn];

bool cmp(const xiaoming &x1,const xiaoming &x2){
	return x1.z<x2.z;
}
int Getfather(int x){
	if(prt[x]==x){
		return x;
	}
	prt[x] = Getfather(prt[x]);
	return prt[x];
}
void Kruskal(){
	int f1,f2,k;
	k=0;
	for(int i=1;i<=n;i++){
		prt[i]=i;
	}
	for(int i=1;i<=m;i++){
		f1 = Getfather(a[i].x);	
		f2 = Getfather(a[i].y);
		if(f1!=f2){
			prt[f1] = f2;
			k++;
			if(k==n-1){
				cout << a[i].z << endl;
				break;
			}
		}
	}
	
}
int main(){
	cin >> n >> m ;
	for(int i=1;i<=m;i++){
		cin >> a[i].x >> a[i].y >> a[i].z;
	}
	sort(a+1,a+m+1,cmp);
	Kruskal();
	return 0;
}

P2820 局域网

算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
int n,m,s,t;
long long ans = 0;
long long sum = 0;
int prt[maxn];
typedef struct xiaoming{
	int x;
	int y;
	int z;
}xiaoming;
xiaoming a[maxn];

bool cmp(const xiaoming &x1,const xiaoming &x2){
	return x1.z<x2.z;
}
int Getfather(int x){
	if(prt[x]==x){
		return x;
	}
	prt[x] = Getfather(prt[x]);
	return prt[x];
}
void Kruskal(){
	int f1,f2,k;
	k=0;
	for(int i=1;i<=n;i++){
		prt[i]=i;
	}
	for(int i=1;i<=m;i++){
		f1 = Getfather(a[i].x);	
		f2 = Getfather(a[i].y);
		if(f1!=f2){
			ans = ans +a[i].z;
			prt[f1] = f2;
			k++;
			if(k==n-1){
				cout << sum-ans << endl;
				break;
			}
		}
	}
	
}
int main(){
	cin >> n >> m ;
	for(int i=1;i<=m;i++){
		cin >> a[i].x >> a[i].y >> a[i].z;
		sum+=a[i].z;
	}
	sort(a+1,a+m+1,cmp);
	Kruskal();
	return 0;
}

P2504 [HAOI2006]聪明的猴子

算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal + 二维坐标距离

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+5;
int a[505];
int father[1005];
int cnt = 0;
struct node{
	int x;
	int y;
	double dis;
};
node t[maxn];
struct input{
	int x;
	int y;
};
input in[1005];

bool cmp(node t1,node t2){
	return t1.dis < t2.dis;
}
int findfather(int x){
	if(x == father[x]){
		return x;
	}
	int F = findfather(father[x]);
	father[x] = F;
	return F;
}
void Union(int x,int y){
	int fx = findfather(x);
	int fy = findfather(y);
	if(fx!=fy){
		father[fx] = fy;
	}
}
int main(){
	int M;
	cin >> M;
	for(int i=1;i<=M;i++){
		cin >> a[i];
	}
	int N;
	cin >> N;
	for(int i=1;i<=N;i++){
		cin >> in[i].x >> in[i].y;
	}
	for(int i=1;i<=N-1;i++){
		for(int j=i+1;j<=N;j++){
			t[cnt].x = i;
			t[cnt].y = j;
			t[cnt].dis = sqrt(	(in[i].x-in[j].x)*(in[i].x-in[j].x) +
								(in[i].y-in[j].y)*(in[i].y-in[j].y)	);
			cnt++;
		}
	}
	sort(t,t+cnt,cmp);
	for(int i=1;i<=N;i++){
		father[i] = i; 
	}
	int count = 0;
	double ans = 0.0;
	for(int i=0;i<cnt && count<N-1;i++){
		int u = t[i].x;
		int v = t[i].y;
		if(findfather(u) != findfather(v)){
			Union(u,v);
			count++;
			ans = t[i].dis;
		}
	}
	int res = 0;
	for(int i=1;i<=M;i++){
		if(a[i]>=ans){
			res++;
		}
	}
	cout << res << endl;
	return 0;
} 

P1991 无线通讯网

算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal + 二维坐标距离

#include<bits/stdc++.h>
using namespace std;
typedef struct city{
	int x;
	int y;
	double z;
}city;
const int maxn = 1e5*2+5;
int prt[maxn];
city a[maxn];
int x[maxn];
int y[maxn];

int S,P;
double ans = 0.0;
int sum = 1;
	
double dis(int x1,int x2,int y1,int y2){
	return sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2));
}

bool cmp(const city &x,const city &y){
	return x.z < y.z;
}

int Getfather(int x){
	if(prt[x]==x){
		return x;
	}
	prt[x]=Getfather(prt[x]);
	return prt[x];
}
void Kruskal(){
	int f1,f2,k;
	k=0;
	for(int i=1;i<=P;i++){
		prt[i]=i;
	}
	for(int i=1;i<sum;i++){
		f1 = Getfather(a[i].x);
		f2 = Getfather(a[i].y);
		if(f1!=f2){
			ans = a[i].z;
			k++;
			prt[f1]=f2;
			if(k==P-S){
				break;
			}	
		}
	}
	
}

int main(){
	cin >> S >> P;
	for(int i=1;i<=P;i++){
		cin >> x[i] >> y[i];
	}

	for(int i=1;i<=P-1;i++){
		for(int j=i+1;j<=P;j++){
			a[sum].x=i;
			a[sum].y=j;
			a[sum].z = dis(x[i],x[j],y[i],y[j]);
			sum++;
		}
	}
	
	sort(a+1,a+sum+1,cmp);
	
	Kruskal();
	printf("%.2lf\n",ans);
	return 0;
} 

P4826 [USACO15FEB]Superbull S

算法标签: 树,二叉树 + 最大生成树
注意点: 模板题 + kruskal (最大生成树)

#include<bits/stdc++.h>
using namespace std;
const int maxn = 4e6+5;
int a[2005];
int father[2005];
int cnt = 0;
struct node{
	int x;
	int y;
	int dis;
};
node t[maxn];
bool cmp(node t1,node t2){
	return t1.dis > t2.dis;
}
int findfather(int x){
	if(x == father[x]){
		return x;
	}
	int F = findfather(father[x]);
	father[x] = F;
	return F;
}
void Union(int x,int y){
	int fx = findfather(x);
	int fy = findfather(y);
	if(fx!=fy){
		father[fx] = fy;
	}
}
int main(){
	int n;
	cin >> n;
	for(int i=1;i<=n;i++){
		cin >> a[i];
	}
	for(int i=1;i<=n-1;i++){
		for(int j=i+1;j<=n;j++){
			t[cnt].x = i;
			t[cnt].y = j;
			t[cnt].dis = a[i] ^ a[j];
			cnt++;
		}
	}
	for(int i=1;i<=n;i++){
		father[i] = i;
	}
	sort(t,t+cnt,cmp);
	long long ans = 0;
	int count = 0;
	for(int i=0;i<cnt && count<n-1;i++){
		int u = t[i].x;
		int v = t[i].y;
		if(findfather(u)!=findfather(v)){
			Union(u,v);
			count++;
			ans += t[i].dis;
		}
	}
	cout << ans << endl;
	return 0;
} 

P1195 口袋的天空

算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal

#include<bits/stdc++.h>
using namespace std;
int prt[1005];
int N,M,K;

typedef struct Edge{
	int x;
	int y;
	int z;
}Edge;
Edge a[10005];

bool cmp(const Edge &x,const Edge &y){
	return x.z< y.z;
}

long long ans = 0;

int Getfather(int x){
	if(prt[x]==x){
		return x;
	}
	prt[x]=Getfather(prt[x]);
	return prt[x];
}

void Kruskal(){
	int f1,f2,k;
	k=0;
	for(int i=1;i<=N;i++){
		prt[i]=i;
	}
	for(int i=1;i<=M;i++){
		f1 = Getfather(a[i].x);
		f2 = Getfather(a[i].y);
		if(f1!=f2){ 
			ans = ans + a[i].z;
			prt[f1] = f2;
			k++;
			if(k==N-K){
				break;
			}
		}
	}
}
int main(){
	
	cin >> N >> M >> K;
	if(N-M>K){
		cout << "No Answer"<<endl;
		return 0;
	}
	
	for(int i=1;i<=M;i++){
		int x,y,z;
		cin >> x >> y >> z;
		a[i].x = x;
		a[i].y = y;
		a[i].z = z;
	}
	sort(a+1,a+M+1,cmp);
	Kruskal();
	cout << ans << endl;
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值