team training 5 F 最小生成树

最新推荐文章于 2025-08-17 22:36:04 发布

转载最新推荐文章于 2025-08-17 22:36:04 发布 · 215 阅读

本文通过一道具体的题目探讨了最小生成树(MST)算法的优化方法，并分享了从大佬代码中学到的技巧，如删除边的策略、使用bool变量重用代码等。

这道题很好，，，

我一开始用的是那种苗条生成树（最大边-最小边的））方式，，这样当然是错了。//突然发现，，这样的学习是比较有收获的，，我一定要多做这种学习，。，而且要快‘’

Gym-100803F]

看了大佬的代码，太好了，，学到了很多。。

（1）del的技术

（2）用bool f 来重用代码

（3）删掉边的思想。

（4）

#include<bits/stdc++.h>

using namespace std;
#define sf scanf
#define pf printf
#define mem(a,b) memset(a,b,sizeof(a))
#define bug1 printf("bug1\n");
#define N 505
#define M 50050
#define INF 1e9
int u[M],v[M],w[M],id[M];
int fa[N];
bool del[M],vis[M];

bool cmp(int a,int b){
    return w[a]<w[b];
}
int n,m;
int find(int x){
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
int cal(bool f){//这个f用的，， 衔接性太好了。
    for(int i=1;i<=n;++i)fa[i]=i;
    int ans=0;
    int cnt=0;
    for(int k=0;k<m;++k){
        int i=id[k];
        if(del[i])continue;
        int fu=find(u[i]),fv=find(v[i]);
        if(fu==fv)continue;
        cnt++;
        fa[fu]=fv;
        ans+=w[i];
        if(f)vis[i]=1;
    }
    if(cnt!=n-1)return INF;//这个地方返回INF，自己也是要会用的，，要不然反而小了怎么办。，
    return ans;
}

int main(){
    while(~sf("%d%d",&n,&m)){
        mem(vis,0);
        mem(del,0);
        for(int i=0;i<m;++i){
            sf("%d%d%d",&u[i],&v[i],&w[i]);
            id[i]=i;
        }
        sort(id,id+m,cmp);//这种id的排序方式。。
        for(int i=1;i<=n;++i)fa[i]=0;
        int minn=cal(1);
        int cnt=0,ans=0;
        for(int i=0;i<m;++i)if(vis[i]){//如此来遍历最小生成树的边，
            del[i]=1;
            if(cal(0)>minn){
                cnt++;
                ans+=w[i];
            }
            del[i]=0;
        }
        pf("%d %d\n",cnt,ans);
    }

}