51nod

1640 天气晴朗的魔法

题目来源： 原创

基准时间限制：1 秒 空间限制：131072 KB 分值: 20  难度：3级算法题

 收藏

 关注

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

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

题目的意思就是要求生成树中所有边的权值中，最大值最小，也就是说最大边权尽量小，组成一个生成树；

第二步就是求这个生成树的最大权值和，也就是最大生成树；

那先来说第一步，其实就是按照kruskal算法，将边权按照从小到大排序，然后从小到大枚举，看到哪个权值能够组成一棵树，其实也就是覆盖了所有的点；然后记录这个权值w_max，再次调用kruskal，这次从大到小排序，然后所有大于w_max的都过滤，只使用比这个小的边，然后组成的最大生成树，就是结果；

写了一下午  tle  不知道怎么回事  借鉴大牛的代码  思路基本一样 但还是 超时！！！操  贴下tle 代码  日后改把

#include<iostream>
#include<cstring>
#include<algorithm>
#include<stdio.h>
using namespace std;
#define maxn 10000+10
#define  inf 9999999
#define mem(a,b) memset(a,b,sizeof(a))
struct road{
    int x,y,z;
}a[maxn];
int dis[maxn];
bool cmp(struct road q,struct road w){
      return q.z<w.z;
}
int find1(int x){
   if(x!=dis[x])
      dis[x]=find1(dis[x]);
   return dis[x];
}

int mix(int x,int y){                //并查集
   int x1;int y1;
   x1=find1(x);
   y1=find1(y);

   if(x1!=y1){
       dis[y1]=x1;
       return 1;
   }
   return 0;
}
int main(){
    int n,m;
      cin>>m>>n;
    for(int j=1;j<=n;j++){
        scanf("%d%d%d",&a[j].x,&a[j].y,&a[j].z);
    }
    int sum=0,num=0;
    sort(a+1,a+n+1,cmp);
    int l=0;
    mem(dis,0);
    for(int j=1;j<=n;j++){
       if(l==m){
            l=j;
            break;
       }
       if(!dis[a[j].x]){
           dis[a[j].x]=1;
           l++;
       }
       if(!dis[a[j].y]){
           dis[a[j].y]=1;
           l++;
       }
    }
    while (l < n)
    {
        if (a[l].z == a[l - 1].z)
        {
            l ++;
        }
        else
            break;
    }
    
    for(int j=1;j<=m;j++){
        dis[j]=j;
    }
    sum=0,num=0;
    for(int j=l-1;j>=1;j--){
        if(mix(a[j].x,a[j].y)){
            sum++;
            num+=a[j].z;
        }
        if(sum==m-1)
            break;
    }
    cout<<num<<endl;

    return 0;
}

大牛代码

#include<iostream>
#include<set>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
using namespace std;
#define LL long long
typedef struct NODE
{
    int a,b;
    int v;
}Node;
Node nodes[200005];
bool cmp(const Node &s, const Node &t)
{
    return s.v < t.v;
}
int father[100005];
int rank[100005];
int hash[100005];
int search_x(int x)
{
    //return x == father[x] ? x : search_x(father[x]);
    if (x != father[x])
        father[x] = search_x(father[x]);
    return father[x];
}
bool union_xy(int x, int y)   //我觉得 都是一样的路径压缩啊 不知道是不是这里出了问题
{
    int root1 = search_x(x);
    int root2 = search_x(y);
    if (root1 == root2)
        return false;
    else if (rank[root1] > rank[root2])           //下面都看不懂  好气！ 自己好差劲的说！
    {
        father[root2] = root1;
    }
    else
    {
        father[root1] = root2;
        if (rank[root1] == rank[root2])
        {
            rank[root2]++;
        }
    }
    return true;
}
int main()
{
    int n,m;
    //cin >> n >> m;
    scanf("%d %d", &n, &m);
    for (int i = 0; i < m; ++ i)
    {
        //cin >> nodes[i].a >> nodes[i].b >> nodes[i].v;
        scanf("%d %d %d", &nodes[i].a, &nodes[i].b, &nodes[i].v);
    }
    sort(nodes, nodes+m, cmp);
    set<int> st;
    int pos = 0;
    int sum = 0;
    memset (hash, 0, sizeof(hash));
    for (int i = 0; i < m; ++ i)
    {
        if (sum == n)
        {
            pos = i;
            break;
        }
        if (! hash[nodes[i].a])
        {
            sum ++;
            hash[nodes[i].a] = 1;
        }
        if (! hash[nodes[i].b])
        {
            sum ++;
            hash[nodes[i].b] = 1;
        }

    }
    //注意我这里是找到最大权值的上边界，这样我就可以直接从pos-1的位置求最大生成树了
    //不这样也行，那么在调用kruskal算法时，大于这个最大权值的过滤掉也是一样的
    while (pos < m)
    {
        if (nodes[pos].v == nodes[pos - 1].v)
        {
            pos ++;
        }
        else
            break;
    }
    for (int i = 1; i <= n; ++ i)
    {
        father[i] = i;
        rank[i] = 1;
    }
    LL weight = 0;
    int num = 0;
    for (int i = pos - 1; i >= 0; -- i)
    {
        if (union_xy(nodes[i].a, nodes[i].b))
        {
            weight += nodes[i].v;
            num ++;

        }
        if (num == n - 1)
        {
            //cout << weight <<endl;
            printf("%lld\n", weight);
            break;
        }
    }
    return 0;
}