51nod1640天气晴朗的魔法最小生成树配合最大生成树

1640 天气晴朗的魔法

  1. 1 秒
  2.  
  3. 131,072 KB
  4.  
  5. 20 分
  6.  
  7. 3 级题
这样阴沉的天气持续下去,我们不免担心起他的健康。

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

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

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

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

现在给定魔法师人数N,魔法链数目M。求此魔法阵的最大效果。

 收起

输入

两个正整数N,M。(1 <= N <= 10^5, N <= M <= 2 * 10^5)

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

保证输入数据合法。

输出

输出一个正整数R,表示符合条件的魔法阵的魔力值之和。

输入样例

4 6
1 2 3
1 3 1
1 4 7
2 3 4
2 4 5
3 4 6

输出样例

12
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define maxn 100005
#define maxm 200005
#define ll long long
using namespace std;
int n,m;
int maxw;
struct Edge
{
    int u,v,dist;
    Edge(){}
    Edge(int u,int v,int d):u(u),v(v),dist(d){}

};
 bool cmp1(Edge a,Edge b)
 {
     return a.dist<b.dist;
 }
 bool cmp2(Edge a,Edge b)
 {
     return a.dist>b.dist;
 }
struct Kruskal
{
    int n,m;
    Edge edges[maxm];
    int fa[maxn];
    int findset(int x){ return fa[x]==-1? x:fa[x]=findset(fa[x]); }

    void init(int n)
    {
        this->n=n;
        m=0;
        memset(fa,-1,sizeof(fa));
    }

    void AddEdge(int u,int v,int dist)
    {
        edges[m++]=Edge(u,v,dist);
    }


      int kruskal()
    {

        int cnt=0;

        sort(edges,edges+m,cmp1);

        for(int i=0;i<m;i++)
        {
            int u=edges[i].u, v=edges[i].v;
            if(findset(u) != findset(v))
            {


                fa[findset(u)] = findset(v);
                if(++cnt>=n-1)
                    return edges[i].dist;
            }
        }
       return -1;

    }
      ll kruskal2()
    {
        ll sum=0;
        int cnt=0;
        sort(edges,edges+m,cmp2);

        for(int i=0;i<m;i++)
        {
            int u=edges[i].u, v=edges[i].v;
            if(findset(u) != findset(v)&&edges[i].dist<=maxw)
            {

                sum +=edges[i].dist;
                fa[findset(u)] = findset(v);
                if(++cnt>=n-1)
                    break;
            }
        }
        if(cnt<n-1)
       return -1;
       return sum;

    }
}KK1,KK2;


int main()
{
    int n,m,k;
    while(~scanf("%d%d",&n,&m))
    {
        int a,b,v;
        KK1.init(n);
        KK2.init(n);
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&a,&b,&v);
            KK1.AddEdge(a,b,v);
            KK2.AddEdge(a,b,v);

        }
        maxw=KK1.kruskal();
        ll sum=KK2.kruskal2();
        printf("%lld\n",sum);

    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/sdauguanweihong/article/details/89647170