最小生成树(prim和kruskal)

kruskal算法

  建图之后,首先对边的权值排序,每次取出没有被选中并且最小的边,用并查集判断边的两点是否相连,如果相连则跳过找下一个点,否则连接着两个点,当连接的边数等于点数-1时,就找到了最小生成树。
  例题:Networking POJ - 1287

#include<iostream>
#include<algorithm>
#include<cstdio>
#define max_n 55

using namespace std;
int father[max_n];
struct edge{
    int from,to;
    long long cost;
}e[max_n*max_n];

bool cmp(edge a,edge b)
{
    return a.cost<b.cost;//自定义cmp
}
int findd(int x)
{
    if(x==father[x])return x;
    else
        return father[x]=findd(father[x]);//路径压缩
}

int main()
{
    int p,r;
    while(scanf("%d%d",&p,&r)!=EOF,p)
    {
        long long ans=0;
        for(int i=1;i<=p;i++)
        {
            father[i]=i;//初始化
        }
        for(int i=1;i<=r;i++)
        {
            cin>>e[i].from>>e[i].to>>e[i].cost;
        }
        sort(e+1,e+1+r,cmp);//对边的权值排序
        for(int i=1;i<=r;i++)
        {
            int a=findd(e[i].from);//判断两个点是否已经相连
            int b=findd(e[i].to);
            if(a!=b)//如果没有相连则连起来,ans加上这条边的权值
            {
                ans+=e[i].cost;
                father[a]=b;
            }
        }
        cout<<ans<<endl;
    }
    return 0;
}

prim

  核心思路是贪心,每次找当前没有选中的最小边连接起来,可以用priority_queue每次弹出当前最小的边。
  例题:还是畅通工程 HDU - 1233


#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
typedef long long ll;
int n;
const int maxx =110 ;
const int inf=0x3f3f3f3f;
int mapp[maxx][maxx];//存图,为了省空间也可以用邻接表,
bool visited[maxx];
struct edge{
	int u, v, w;
	edge(int u, int v, int w) {
		this->u = u;
		this->w = w;
		this->v = v;
	}
	edge() {};
	bool operator<(const edge& a)const {//重载小于号,可以建立小根堆
		return a.w < w;
	}

}e[maxx*maxx];
std::priority_queue<edge>q;
ll prim() {
	memset(visited,0,sizeof(visited));
    int cnt=0;
	int ans = 0;
	for (int i = 1; i <= n; i++) {
		if(mapp[1][i]!=inf){
			q.push(edge(1,i,mapp[1][i]));//首先把跟点1相连的边加入优先队列
		}
	}
	visited[1]=1;
	while(!q.empty()){
		edge temp=q.top();//每次弹出当前队列中权值最小的边
		q.pop();
		if(visited[temp.v])continue;//如果这条边的到达点已经被标记过就continue
		visited[temp.v]=1;
		ans+=temp.w;
        cnt++;
        if(cnt==n-1)return ans;//如果加入的边等于点数-1,说明已经找到了最小生成树
		for(int i=1;i<=n;i++){
			if(mapp[temp.v][i]!=inf)//把跟这个点连接的边全部加入队列
			q.push(edge(temp.v,i,mapp[temp.v][i]));
		}		
	}
	return ans;
}

int main() {
	
	while (scanf("%d", &n) != EOF) {
		if(n==0)break;
		memset(mapp,inf,sizeof(mapp));
		for(int i=1;i<=n;i++){
			mapp[i][i]=0;//初始化
		}
		while (!q.empty())q.pop();
		for (int i = 1; i <= n * (n - 1) / 2; i++) {
			int x, y, z;
			scanf("%d%d%d", &x, &y, &z);
			mapp[x][y] = mapp[y][x] = z;
		}
		ll ans = prim();
		for (int i = 1; i <= n; i++) {
			if (!visited[i])ans = -1;
		}
		if (ans == -1)printf("?\n");
		else
			printf("%lld\n", ans);
	}
	return 0;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值