2022.02.16学习总结（prim，kruskal算法（最小生成树））

最新推荐文章于 2023-02-13 21:17:02 发布

sun*san

最新推荐文章于 2023-02-13 21:17:02 发布

阅读量268

点赞数

文章标签：算法学习数据结构

本文链接：https://blog.csdn.net/m0_63263973/article/details/122972878

版权

视频：prim，kruskal算法理解

prim和kruskal算法比较

kruskal是直接将所有边权重排序后再生成最小树,prim是要多次寻找最小权重的邻边,所以在时间上kruskal比prim更快。

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出 orz。

输入格式

第一行包含两个整数 N,MN,M，表示该图共有 NN 个结点和 MM 条无向边。

接下来 MM 行每行包含三个整数 X_i,Y_i,Z_iXi,Yi,Zi，表示有一条长度为 Z_iZi 的无向边连接结点 X_i,Y_iXi,Yi。

输出格式

如果该图连通，则输出一个整数表示最小生成树的各边的长度之和。如果该图不连通则输出 orz。

输入输出样例

输入 #1复制
4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3
输出 #1复制
7
说明/提示

数据规模：

对于 20\%20% 的数据，N\le 5N≤5，M\le 20M≤20。

对于 40\%40% 的数据，N\le 50N≤50，M\le 2500M≤2500。

对于 70\%70% 的数据，N\le 500N≤500，M\le 10^4M≤104。

对于 100\%100% 的数据：1\le N\le 50001≤N≤5000，1\le M\le 2\times 10^51≤M≤2×105，1\le Z_i \le 10^41≤Zi≤104。

样例解释：

所以最小生成树的总边权为 2+2+3=72+2+3=7。

用kruskal写的模板

#include<bits/stdc++.h>
using namespace std;
const int MAXM=200001;
const int MAXN=5001;
int n,m;//顶点个数，边的条数
long long len;//最小生成树总长
int total;
struct node
{
    int b;//起点
    int e;//终点
    int w;//边权
}edge[MAXM];//创建边集
int par[MAXN];//用于判断是否形成闭环，表示连接该点的下一个点，类似于并查集
bool cmp(node a,node b)//依据权重将边集进行从小到大排序
{
    return a.w<b.w;
}
int Find(int *par,int f)//查找该顶点连接的尾部下标
{
    while(par[f]>0){
        f=par[f];
    }return f;
}
void Kruskal()//kruskal算法
{
    sort(edge+1,edge+m+1,cmp);//排序
    int x,y;
    for(int i=1;i<=m;i++){//循环每一条边
        x=Find(par,edge[i].b);
        y=Find(par,edge[i].e);
        if(x!=y){//若x=y则表示两个点连接的尾部相同，加上当前这条边则形成闭环
            par[x]=y;//将结尾顶点的下标放入起点顶点的par数组值中，表示已存入最小生成树中
            len+=edge[i].w;
            total++;
        }
        if(total==n-1) break;
    }
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=m;i++){
        int p,q,r;
        cin>>p>>q>>r;
        edge[i].b=p;
        edge[i].e=q;
        edge[i].w=r;
    }
    Kruskal();
   if(total<n-1) cout<<"orz"<<endl;
    else cout<<len<<endl;
}

题目背景

还记得 NOIP 2011 提高组 Day1 中的铺地毯吗？时光飞逝，光阴荏苒，三年过去了。组织者精心准备的颁奖典礼早已结束，留下的则是被人们踩过的地毯。请你来解决类似于铺地毯的另一个问题。

题目描述

会场上有 n 个关键区域，不同的关键区域由 m 条无向地毯彼此连接。每条地毯可由三个整数 u、v、w 表示，其中 u 和 v 为地毯连接的两个关键区域编号，w 为这条地毯的美丽度。

由于颁奖典礼已经结束，铺过的地毯不得不拆除。为了贯彻勤俭节约的原则，组织者被要求只能保留 K 条地毯，且保留的地毯构成的图中，任意可互相到达的两点间只能有一种方式互相到达。换言之，组织者要求新图中不能有环。现在组织者求助你，想请你帮忙算出这 K 条地毯的美丽度之和最大为多少。

输入格式

第一行包含三个正整数 n、m、K。

接下来 m 行中每行包含三个正整数 u、v、w。

输出格式

只包含一个正整数，表示这 K 条地毯的美丽度之和的最大值。

输入输出样例

输入 #1复制
5 4 3
1 2 10
1 3 9
2 3 7
4 5 3
输出 #1复制
22
说明/提示

选择第 1、2、4 条地毯，美丽度之和为 10 + 9 + 3 = 22。

若选择第 1、2、3 条地毯，虽然美丽度之和可以达到 10 + 9 + 7 = 26，但这将导致关键区域 1、2、3 构成一个环，这是题目中不允许的。

1<=n,m,k<=100000

#include<bits/stdc++.h>
using namespace std;
int n,m,k;
long long len;
const int MaxNM=100001;
int total;
struct node
{
    int b;
    int e;
    int w;
}edge[MaxNM];
int par[MaxNM];
int Find(int *par,int f)
{
    while(par[f]>0){
        f=par[f];
    }return f;
}
bool cmp(node a,node b)
{
    return a.w>b.w;//按权重由大到小排列
}
void Kru()
{
    sort(edge+1,edge+m+1,cmp);
    int x,y;
    for(int i=1;i<=m;i++){
        x=Find(par,edge[i].b);
        y=Find(par,edge[i].e);
        if(x!=y){
            par[x]=y;
            len+=edge[i].w;
            total++;
        }
        if(total>=k) break;//地毯数目大于等于k则结束
    }
}
int main()
{
    cin>>n>>m>>k;
    for(int i=1;i<=m;i++){
        int z,x,c;
        cin>>z>>x>>c;
        edge[i].b=z;
        edge[i].e=x;
        edge[i].w=c;
    }
    Kru();
    cout<<len<<endl;
}

题目背景

小杉坐在教室里，透过口袋一样的窗户看口袋一样的天空。

有很多云飘在那里，看起来很漂亮，小杉想摘下那样美的几朵云，做成棉花糖。

题目描述

给你云朵的个数 NN，再给你 MM 个关系，表示哪些云朵可以连在一起。

现在小杉要把所有云朵连成 KK 个棉花糖，一个棉花糖最少要用掉一朵云，小杉想知道他怎么连，花费的代价最小。

输入格式

第一行有三个数 N,M,KN,M,K。

接下来 MM 行每行三个数 X,Y,LX,Y,L，表示XX云和 YY 云可以通过 LL 的代价连在一起。

输出格式

对每组数据输出一行，仅有一个整数，表示最小的代价。

如果怎么连都连不出 KK 个棉花糖，请输出 No Answer。

输入输出样例

输入 #1复制
3 1 2
1 2 1
输出 #1复制
1
说明/提示

对于 30\%30% 的数据，N \le 100N≤100，M \le 10^3M≤103；

对于 100\%100% 的数据，1 \le N \le 10^31≤N≤103，1 \le M \le 10^41≤M≤104，1 \le K \le 101≤K≤10，1 \le X,Y \le N1≤X,Y≤N，0 \le L<10^40≤L<104。

棉花糖的数目我们可以理解成可以生成的树的数目，则边为n-k时，生成了k棵树。

#include<bits/stdc++.h>
using namespace std;
int n,m,k;
long long len;
const int MaxN=1001;
const int MaxM=10001;
int total;
struct node
{
    int b;
    int e;
    int w;
}edge[MaxM];
int par[MaxN];
int Find(int *par,int f)
{
    while(par[f]>0){
        f=par[f];
    }return f;
}
bool cmp(node a,node b)
{
    return a.w<b.w;
}
void Kru()
{
    sort(edge+1,edge+m+1,cmp);
    int x,y;
    for(int i=1;i<=m;i++){
        x=Find(par,edge[i].b);
        y=Find(par,edge[i].e);
        if(x!=y){
            par[x]=y;
            len+=edge[i].w;
            total++;
        }
        if(total>=n-k) break;//生成k棵树时，循环结束
    }
}
int main()
{
    cin>>n>>m>>k;
    for(int i=1;i<=m;i++){
        int z,x,c;
        cin>>z>>x>>c;
        edge[i].b=z;
        edge[i].e=x;
        edge[i].w=c;
    }
    Kru();
    if(total<n-k){cout<<"No Answer"<<endl;}
    else cout<<len<<endl;
}