51 nod 1640天气晴朗的魔法(最小生成树)

这样阴沉的天气持续下去,我们不免担心起他的健康。

51nod魔法学校近日开展了主题为“天气晴朗”的魔法交流活动。

N名魔法师按阵法站好,之后选取N - 1条魔法链将所有魔法师的魔力连接起来,形成一个魔法阵。

魔法链是做法成功与否的关键。每一条魔法链都有一个魔力值V,魔法最终的效果取决于阵中所有魔法链的魔力值的和。

由于逆天改命的魔法过于暴力,所以我们要求阵中的魔法链的魔力值最大值尽可能的小,与此同时,魔力值之和要尽可能的大。

现在给定魔法师人数N,魔法链数目M。求此魔法阵的最大效果。
Input
两个正整数N,M。(1 <= N <= 10^5, N <= M <= 2 * 10^5)

接下来M行,每一行有三个整数A, B, V。(1 <= A, B <= N, INT_MIN <= V <= INT_MAX)

保证输入数据合法。
Output
输出一个正整数R,表示符合条件的魔法阵的魔力值之和。
Input示例
4 6
1 2 3
1 3 1
1 4 7
2 3 4
2 4 5
3 4 6
Output示例

12

分析:先按照Kruscal算法求出生成最小成树过程中,记录生成树的最大边。这条边是能构成生成树最低要求的边,记作maxw,符合题目意思了,然后求最大生成树,求法与最小生成树的过程相反,将大于maxw的边剔除,从大到小取边加入边集合,直到所有点都到生成树集合中。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
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(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);
    return 0;
}

/*
4 6
1 2 3
1 3 1
1 4 7
2 3 4
2 4 5
3 4 6
*/


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值