51nod 1640 天气晴朗的魔法 克鲁斯卡尔

题目链接:https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1640
思路:题目要求求最大生成树,并且对最大边有限制,先按照克鲁斯卡尔算法求出生成最小成树过程中,加入生成树的最大边。这条边是能构成生成树最低要求的边,记作maxw,符合题目意思了,然后求最大生成树,求法与最小生成树的过程相反,将大于maxw的边剔除,从大到小取边加入边集合,知道所有点都到生成树集合中。
一开始用个-1标记最大生成树不存在,生成树的权值可以为负,一直错在一个样例上。还有就是并查集竟然忘记路径压缩(╥╯^╰╥)

#include<cstdio>// 最大生成树 
#include<queue>
#include<iostream>
#include<vector>
#include<map>
#include<cstring>
#include<string>
#include<set>
#include<stack>
#include<algorithm>
#define cle(a) memset(a,0,sizeof(a))
#define inf(a) memset(a,0x3f,sizeof(a))
#define ll long long
#define Rep(i,a,n) for(int i=a;i<=n;i++)
using namespace std;
const int INF = ( 2e9 ) + 2;
const ll maxn = 2e5+10;
ll maxw,flag;
struct edge
{
    int u,v;
    ll w;
};
vector<edge> e;
int f[maxn];
bool cmp(edge a,edge b)
{
    return a.w<b.w;
}
int Find(int x)
{
    return f[x]==x?x:f[x]=Find(f[x]);
}
ll Kruscal_1(int n)
{
    ll ans=0;
    for(int i=1;i<=n;i++)f[i]=i;
    sort(e.begin(),e.end(),cmp);
    for(int i=0,L=e.size();i<L;i++)
    {
        ll w=e[i].w;
        int u=e[i].u,v=e[i].v;
        int xx=Find(u);
        int yy=Find(v);
        if(xx!=yy)
        {

            ans+=w;
            f[xx]=yy;
            if(maxw<w)maxw=w;
        }
    }
    return ans;
}
ll Kruscal_2(int n,int maxw)
{
    ll num=0,ans=0;
    for(int i=1;i<=n;i++)f[i]=i;
    for(int i=e.size()-1;i>=0;i--)
    {
        ll w=e[i].w;
        if(w>maxw)continue;
        int u=e[i].u,v=e[i].v;
        int xx=Find(u);
        int yy=Find(v);
        if(xx!=yy)
        {
            ans+=w;
            num++;
            f[xx]=yy;
            if(maxw<w)maxw=w;
        }
    }
    if(num!=n-1) flag = 1;
    else return ans;
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++)
    {
        int u,v;
        ll w;
        scanf("%d%d%lld",&u,&v,&w);
        e.push_back(edge{u,v,w});
    }
    maxw=-1,flag=0;

    ll ans1=Kruscal_1(n);
    ll ans2=Kruscal_2(n,maxw);
    if(flag)printf("%lld\n",ans1);
    else
    printf("%lld\n",ans2);
}
/*
4 6
1 2 3
1 3 1
1 4 7
2 3 4
2 4 5
3 4 6
*/

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值