51NOD--1640 天气晴朗的魔法--二分+最小生成树

1640 天气晴朗的魔法

1. 1.0 秒
2.  
3. 131,072.0 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<bits/stdc++.h>
using namespace std;
#define ll long long
const ll maxn=200000+66;
const ll mod=1e9+7;
int n,m;
struct node
{
    int u;
    int v;
    int w;
} a[maxn];
bool cmp(const node&a,const node&b)
{
    return a.w<b.w;
}
int f[maxn];
int finds(int x)
{
    return f[x]==x?x:f[x]=finds(f[x]);
}
void unions(int x,int y)
{
    int fx=finds(x);
    int fy=finds(y);
    if(fx!=fy)
    {
        f[fx]=fy;
    }
}
ll judge(int mid)
{
    int ans=0;
    ll sum=0;
    for(int i=m; i>=1; i--)
    {
        if(a[i].w>mid)continue;
        int u=a[i].u;
        int v=a[i].v;
        int fu=finds(u);
        int fv=finds(v);
        if(fu!=fv)
        {
           // cout<<u<<" "<<v<<endl;
            f[fu]=fv;
            ans++;
            sum+=a[i].w;
        }
        if(ans==n-1)
            break;
    }
    if(ans==n-1)
        return sum;
    else
        return 0;
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=1; i<=n; i++)
            f[i]=i;
        for(int i=1; i<=m; i++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            a[i].u=u;
            a[i].v=v;
            a[i].w=w;
        }
        sort(a+1,a+m+1,cmp);
        //cout<<judge(5)<<endl;
        //cout<<judge(4)<<"  "<<judge(6)<<"  "<<judge(7);
        int l=1,r=m;
        int mid=(l+r)>>1;
        ll ans=0;
        while(l<=r)
        {
            for(int i=1; i<=n; i++)
                f[i]=i;
            mid=(l+r)>>1;
           ll flag=judge(a[mid].w);
            //cout<<flag<<" "<<a[mid].w<<"  "<<mid<<endl;
            if(flag)
            {
               // cout<<"---";
                r=mid-1;
               // ans=max(ans,flag);
               ans=flag;
            }
            else
            {
                l=mid+1;
            }
            //cout<<l<<"---"<<r<<"  "<<mid<<endl;
        }
        printf("%lld\n",ans);
    }
}