P1547 Out of Hay

Posted 2021-01-03 lck-lck

传送门     练习

只是一个最小生成树的水题，拿来练练模板

AC代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
#define maxn 10001
struct hh
{
    int x,y,z;
}t[maxn];
int n,m,fa[maxn],ans=0,bj,jl[maxn],top=0;
bool cmp(const hh&x,const hh&y)
{
    return x.z<y.z;
}
inline int find(int u)
{
    if(fa[u]==u) return u;
    fa[u]=find(fa[u]);
    return u=fa[u];
}
inline void kruskal()
{
    int f1,f2,k,i;
    k=0;
    for(i=1;i<=n;i++) fa[i]=i;//初始化
    for(i=1;i<=m;i++)
    {
        f1=find(t[i].x);
        f2=find(t[i].y);
        if(f1!=f2)
        {
            ans=ans+t[i].z;
            jl[++top]=t[i].z;
            fa[f1]=f2;
            k++;
            if(k==n-1) break;
        }
    } 
}
int main()
{
    cin>>n>>m;
    ans=0;bj=1;
    for(int i=1;i<=m;i++)
    {
        cin>>t[i].x>>t[i].y>>t[i].z;
    }
    sort(t+1,t+m+1,cmp);
    kruskal();
    sort(jl+1,jl+top);
    cout<<jl[top]<<endl;
return 0;
}

再附上最小生成树纯模板：

#include<bits/stdc++.h>
using namespace std;
const int maxn=200005;
struct edge{int x,y,z;}a[maxn];//定义数组（好用的结构体） 
int n,m,fa[maxn],ans=0,bj;
bool cmp(const edge&x,const edge&y)//重定义sort函数（背模板吧。。） 
{
    return x.z<y.z;
}
int find(int u)//查找祖先 
{
    if(fa[u]==u) return fa[u];
    fa[u]=find(fa[u]);
    return fa[u];
}
void kruskal()//kruskal的核心程序 
{
    int f1,f2,k,i;
    k=0;
    for(i=1;i<=n;i++) fa[i]=i;
    for(i=1;i<=m;i++) 
    {
        f1=find(a[i].x);
        f2=find(a[i].y);//并查集基本操作 
        if(f1!=f2)//两点之间没有边（有边就不管） 
        {
            ans=ans+a[i].z;
            fa[f1]=f2;
            k++;//搭边并记录 
            if(k==n-1) break;//图刚好为最小生成树，退出 
        }
    }
    if(k<n-1){cout<<"orz"<<endl;bj=0;return;}//边数达不到最小生成树要求 
}
int main()
{
    cin>>n>>m;//输入点数，边数 
    ans=0;bj=1;//ans记录总的边权长，bj判断能否组成最小生成树 
    for(int i=1;i<=m;i++)
    {
        cin>>a[i].x>>a[i].y>>a[i].z;//输入点1，点2，并赋上权值 
    }
    sort(a+1,a+m+1,cmp);//从小到大排序 
    kruskal();//建树 
    if(bj) cout<<ans<<endl;//有最小生成树就输出ans 
return 0;
}