2021-01-21

今天在B站上学习了最小生成树的prim和kruskal两种算法的概念
然后去洛谷找题做

用了kruskal算法进行解题 有了思路和想出核心代码的模式 还在讨论区找到一篇对我理解帮助很大的详细题解

#include<bits/stdc++.h>
using namespace std;
int n,m,ans=0,father[5001];
struct fkt{int x;int y;int value;}a[200001];
bool cmp(fkt xx,fkt yy){//首先Kruscal算法的经典思路有点类似贪心，我们需要每次选取最优的边扔到我们的集合里
    //所以我们用到了并查集来作为集合的存放方式，先排序来类似于贪心地选取最小的边
    //肯定最小的边是最优的，∵必然存在N-1条边，不然没意义啊，这题数据还是比较水，没有输出orz的例子，所以我的代码也没有判断orz。。。 
    return xx.value<yy.value;
}
int find_fa(int x){//并查集之找爸爸（简略跳过） 
    if(father[x]!=x) 
        father[x]=find_fa(father[x]);    //我就是在这里翻车的啊QAQ    大家记住
                                        //找爸爸的时候一定要是father[x]=find_fa(father[x])
                                        //找爸爸的时候一定不要father[x]=find_fa(x) ∵这样你的函数等于一直无限循环下去，导致MLE的结局TAT 
    return father[x];
}
void unionn(int x,int y){//并查集之合并（简略跳过） 
    x=find_fa(x);
    y=find_fa(y);
    father[y]=x;
    return;
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)    father[i]=i;//并查集之初始化 
    for(int i=1;i<=m;i++)//读入 
        scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].value);
    sort(a+1,a+m+1,cmp);//贪心排序 
    n--;//我们最多只会有N-1条边就可以最小生成树 
    for(int i=1;i<=m;i++){//开始找，Kruskal运用的是找边的方式 
        int xxx=find_fa(a[i].x),yyy=find_fa(a[i].y);//首先判断我这两个点有没有在同一集合内（如果在同一集合内说明这两个点必然存在一条路径可以联通） 
        if(xxx!=yyy){//如果不连通就把它连起来 
            ans+=a[i].value;//记录答案 
            unionn(a[i].x,a[i].y);//合并两个点，说明这两个点已经在同一集合内了 
            n--;//满足一条我们删一次N 
        }
        if(n==0)break;//找到了就退出 
    }
    printf("%d\n",ans);//输出结果 
    return 0;
}