2021牛客寒假算法基础集训营4 F 并查集 贪心 思维建边

题目

题解思路

贪心
能拿提升能力大就必须拿，先不考虑给谁。
当相同关系的武器但提升能力少的时候，肯定是拿大的。

思维建图
将武器的分配看成一种关系，即 将两个人连接的桥梁（单人的时候自环即可）。
利用并查集让两个人组成一个连通块。即装备可以在这两个人之间选择。

当有新装备加入的时候，将连通块扩大，即N-1个装备，肯定在这N个人中选择N-1个人拿装备。有1个人将拿不到装备。
当并查集出的两个人已经在同一个连通块中时，即N个人都有装备了，在连通图中表现为自环连通块。当有其他连通块与自环的连通的时候，其他连通块上少装备的那个人就可以有装备了。即连接起来也标记为自环。
这样我们只需贪心的将武器提升的能力从大到小排序，再将他带来的关系 是否成为与连通块的部分 作为是否使用了它的依据即可。

这样建边属实巧妙。

AC代码

#include <bits/stdc++.h>
//#include <unordered_map>
//priority_queue
#define PII pair<int,int>
#define ll long long

using namespace std;

const  int  INF =  0x3f3f3f3f;
int n , m ;
const  int N = 200100 ; 
vector <int> head[N] ; 
struct node 
{
    int w , x , y ;
}s[N];
int a[N] ; 
int vis[N] ; 
int find(int x )
{
    if ( x != a[x] )
        return a[x] = find(a[x]) ;
    return a[x] ;
}
bool cmp (node A , node B )
{
    return A.w > B.w ; 
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    cin >> n >> m ;
    long long ans = 0 ; 
    for (int i = 1 ; i <= n ; i++ )
    {
        a[i] = i ; 
    }
    for (int i = 1 ; i <= m ; i++ )
    {
        int k , t1 , t2 ;
        cin >>  k ;
        if ( k == 1 )
        {
            cin >> t1 ;
            t2 = t1 ;
        }else
        {
            cin >> t1 >> t2 ; 
        }
        int sp ; 
        cin >> sp ; 
        s[i] = {sp,t1,t2} ; 
    }
    sort(s+1,s+1+m,cmp);
    for (int i = 1 ; i <= m ; i++ )
    {
        int fx = find(s[i].x) , fy = find(s[i].y) ;
        if ( fx != fy && (!vis[fx]||!vis[fy]))
        {
            ans += s[i].w ; 
            a[fy] = fx ;
            vis[fx] = vis[fx]|vis[fy] ; 
        }else if ( !vis[fx] ) 
        {
            ans += s[i].w ;
            vis[fx] = 1 ; 
        }
    }
    cout << ans << "\n" ; 
    return 0 ;
}