Kruskal/prim--最小生成树

问题概述:已知有n个村庄以及部分村庄之间的所有公路长度,你的目标是以最短的公路使全省任意两个村庄间都可以实现公路交通,而其他公路全部废弃,求出满足条件的最短公路长度,如果没法联通,输出-1

输入样例:                                 对应输出:

3 7                                              16

1 2 19

2 3 11

3 1 7

1 3 5

2 3 89

3 1 91

1 2 32


Kruskal:

套路:按边从短到长排序,一条边一条边的往图中加,如果加完这条边之后原图中出现环,则舍弃这条边,最后形成的就是最小生成树了

原理与证明:http://www.cnblogs.com/biyeymyhjob/archive/2012/07/30/2615542.html

复杂度:O(mlogm)

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int ufs[1205];
typedef struct
{
    int u;
	int v;
	int len;
}Road;
Road s[22222];
int Find(int x);
bool comp(Road a, Road b)
{
	if(a.len<b.len)
		return 1;
	return 0;
}
int main(void)
{
    int n, i, t1, t2, sum, m, flag;
	while(scanf("%d%d", &n, &m)!=EOF)
	{
		for(i=1;i<=m;i++)
			scanf("%d%d%d", &s[i].u, &s[i].v, &s[i].len);
		for(i=1;i<=n;i++)
			ufs[i] = -1;
		flag = n-1;
		sum = 0;
		sort(s+1, s+m+1, comp);
		for(i=1;i<=m;i++)   /*使用并查集*/
		{
			t1 = Find(s[i].u);
			t2 = Find(s[i].v);          /*如果这个公路没有必要,也就是u和v这两个村庄已经被联通了,就不修这条路*/
			if(t1!=t2)
			{
				flag--;
				sum += s[i].len;
				ufs[t1] = t2;          /*将这几个村庄依次连接*/
			}
		}
		if(flag==0)
			printf("%d\n", sum);
		else
			printf("-1\n");
	}
    return 0;
}

int Find(int x)
{
    if(ufs[x]==-1)
		return x;
    return ufs[x] = Find(ufs[x]);
}


Prim:

套路:初始化所有点都没有标记,然后标记1号点

每次找一条其中一端已经被标记而另一端没被标记的当前最短的边加入图中并标记另一端,直到所有点都被标记

复杂度:O(nm)

#include<stdio.h>
#include<string.h>
#include<limits.h>
#include<math.h>
int path[102][102] = {0}, flag[102] = {0}, closedge[102] = {0};
int sum;
int CreatMST(int n);
int main()
{
    int i, x, y, n, m;
    scanf("%d", &n);
	sum = 0;
    for(i=1;i<=n*(n-1)/2;i++)
    {
        scanf("%d%d%d", &x, &y, &m);
        path[x][y] = path[y][x] = m;       /*path[x][y]表示x城市到y城市的距离*/
    }
    for(i=1;i<=n;i++)
        path[i][i] = INT_MAX;         /*显然不可能修一条从自己城市到自己城市的路,直接设为max*/
    printf("%d\n", CreatMST(n));
    return 0;
}

int CreatMST(int n)
{
    int i, j, x, k;
    flag[1] = 1;                      /*flag[n]==1表示第n个城市已经被连起来了,而flag[n]==0表示这个城市还是独立的*/
    for(i=2;i<=n;i++)
		closedge[i] = path[1][i];
    for(i=2;i<=n;i++)
    {
        k = INT_MAX, x = 2;
        for(j=2;j<=n;j++)
		{
			if(flag[j]==0 && closedge[j]<k)      /*看还没有修的哪一条路最短,先修那条路*/
			{
				x = j;
				k = closedge[j];
			}
		}
        flag[x] = 1;              /*第x个城市连接完毕*/
        sum += k;                 /*加上要修的路长度(最小值)*/
        for(j=2;j<=n;j++)
        {
			if(flag[j]==0 && path[x][j]<closedge[j])           /*找到已经修好路的城市到j城市最短的路*/
				closedge[j] = path[x][j];
		}
    }
    return sum;
}


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值