最小生成树

Kruskal算法

朴素版本:

1.用贪心的思想排序
2.用并查集判断边是否可选
3.用ans累加答案

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>

using namespace std;

const int maxn=100010;
const int maxm=200010;

int n,m,f[maxn];//f是并查集

struct edge
{
	int s,e,d;//起点 终点 长度
}ed[maxm];

bool cmp(edge a,edge b)
{
	return a.d<b.d;
}

int get(int p)
{
	if (f[p]==p) return p;
	else return f[p]=get(f[p]);
}

int main()
{
	scanf("%d%d",&n,&m);
	for (int a=1;a<=m;a++)
		scanf("%d%d%d",&ed[a].s,&ed[a].e,&ed[a].d);
	sort(ed+1,ed+m+1,cmp);
	for (int a=1;a<=n;a++)
		f[a]=a;
	
	int ans=0;
	for (int a=1;a<=m;a++)
	{
		int s=ed[a].s,e=ed[a].e,d=ed[a].d;
		int f1=get(s);
		int f2=get(e);
		if (f1!=f2)
		{
			f[f1]=f2;
			ans+=d;
			z[++k]=a;
		}
	}
	printf("%d\n",ans);

	return 0;
}

Prim算法

例题:

农民约翰被选为他们镇的镇长!他其中一个竞选承诺就是在镇上建立起互联网,并连接到所有的农场。当然,他需要你的帮助。

约翰已经给他的农场安排了一条高速的网络线路,他想把这条线路共享给其他农场。为了用最小的消费,他想铺设最短的光纤去连接所有的农场。

你将得到一份各农场之间连接费用的列表,你必须找出能连接所有农场并所用光纤最短的方案。

每两个农场间的距离不会超过100000

输入:

第一行: 农场的个数,N(3<=N<=100)。
第二行…结尾: 后来的行包含了一个N*N的矩阵,表示每个农场之间的距离。理论上,他们是N行,每行由N个用空格分隔的数组成,实际上,他们限制在80个字符,因此,某些行会紧接着另一些行。当然,对角线将会是0,因为不会有线路从第i个农场到它本身。

样例:
输入:

4

0 4 9 21

4 0 8 17

9 8 0 16

21 17 16 0

输出:

28

原理:

建立一个点集S,里面装着已经有边连接的点(初始只有节点1)
找出一条边,使它连接的端点恰好有一个点在S里且权值最小
答案ans+=这条边的权值
将另一个端点放进S里
计数器cnt++
当cnt==n(点数)-1时退出算法,输出ans

代码(原始版本,时间复杂度O(n^3)):

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int n;
int a[110][110];
bool vis[110];
int b[110];
long long ans;
int o;
int main(){
	scanf("%d",&n);
	for (int i=1;i<=n;i++){
		for (int j=1;j<=n;j++){
			scanf("%d",&a[i][j]);
		}
	}
	int s=1;
	vis[s]=true;
	b[++o]=s;
	for (int i=1;i<n;i++){
		int f1=0,f2=0,sum=114514;
		for (int j=1;j<=o;j++){
			for (int k=1;k<=n;k++){
				if (b[j]!=k&&vis[k]==false&&sum>a[b[j]][k]){
					sum=a[b[j]][k];
					f1=b[j];
					f2=k;
				}
			}
		}
		ans+=sum;
		vis[f2]=true;
		b[++o]=f2;
	}
	printf("%lld",ans);
	return 0;
}

代码(正常版本,时间复杂度O(n^2+m)):

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n;
long long a[1100][1100],ans;
long long dis[1100];
bool vis[1100];
int main(){
	scanf("%d",&n);
	for (int i=1;i<=n;i++){
		for (int j=1;j<=n;j++){
			scanf("%lld",&a[i][j]);
			if (i==j){
				a[i][j]=1145141919810;
			}
		}
		dis[i]=1145141919810;
	}
	vis[1]=true;
	dis[1]=0;
	for (int i=2;i<=n;i++){
		dis[i]=min(dis[i],a[1][i]);
	}
	for (int i=1;i<n;i++){
		long long sum=1145141919810;
		int f=0;
		for (int j=1;j<=n;j++){
			if (vis[j]==false&&dis[j]<sum){
				sum=dis[j];
				f=j;
			}
		}
		vis[f]=true;
		ans+=sum;
		for (int j=1;j<=n;j++){
			if (j!=f)
			dis[j]=min(dis[j],a[f][j]);
		}
	}
	long long sum=0;
	printf("%lld",ans);
	return 0;
}
顺带提一嘴,添加虚点是常用伎俩

两个算法的优化版本将在之后更新……

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值