51Nod - 1640 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

这道题的题意是首先满足最大值尽可能小,然后再满足魔力值和尽可能的大;

首先跑一遍最小生成树找出最大的魔力值。然后对所有小于这条边的边跑一边最大生成树。

最大生成树和最小生成树的过程是一样的,只是排序不一样而已。


#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define N 100005
struct edge
{
    int u;
    int v;
    long long w;
};
struct edge e[2*N];
int f[N];
int n,m;
int cmp1(edge x,edge y)
{
    return x.w<y.w;
}
int cmp2(edge x,edge y)
{
    return x.w>y.w;
}
int getf(int a)
{
    if(f[a]==a)
        return a;
    else
    {
        f[a]=getf(f[a]);
        return f[a];
    }
}
int F(int u,int v)
{
    int t1,t2;
    t1=getf(u);
    t2=getf(v);
    if(t1!=t2)
    {
        f[t2]=t1;
        return 1;
    }
    return 0;
}
int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    long long sum=0;
    for(int i=1; i<=m; i++)
    {
        scanf("%d %d %lld",&e[i].u,&e[i].v,&e[i].w);
    }
    sort(e+1,e+m+1,cmp1);
    for(int i=0; i<=n; i++)
        f[i]=i;
    int countt=0;
    long long maxx=-9999999999;//注意有负值的存在;
    for(int i=1; i<=m; i++)
    {
        if(F(e[i].u,e[i].v))
        {
            if(maxx<=e[i].w)//找出最大的权边
                maxx=e[i].w;
            countt++;
        }
        if(countt==n-1)
            break;
    }
    //printf("%d\n",maxx);
    for(int i=0; i<=n; i++)
        f[i]=i;
    sort(e+1,e+1+m,cmp2);
    int count1=0;
    for(int i=1; i<=m; i++)
    {
        if(e[i].w<=maxx)//对于比最大边的所有小的边跑最大生成树
        {
            if(F(e[i].u,e[i].v))
            {
                count1++;
                sum+=e[i].w;
            }
        }
        if(count1==n-1)
            break;
    }
    printf("%lld\n",sum);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_41925919/article/details/80268571