牛客 魏迟燕的自走棋

题目链接：https://ac.nowcoder.com/acm/contest/9984/F

思路：首先对于k=2  每一对人合并成一个集合，表示这个集合内都可以拿起某件装备

对不同的关系都可以合并集合，因为在集合中的任何人都可以拿起 这些装备，通过传递即可

那么再考虑贪心， 对于某件更有价值的装备肯定是要优先拿的，那么并查集维护集合，并且用vis判断集合

内的人是否已经拿满不能够再拿装备了即可

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
const int mod=1e9+7;
#define ll long long
#define ull unsigned long long
#define pi pair<int,int>
#define fi first
#define sc second
#define pb push_back
int n,m;
int f[maxn];
int vis[maxn];

struct ac
{
    int u,v,w;
    bool operator<(ac a)
    {
        return w>a.w;
    }
};
ac a[maxn];

int find1(int x)
{
    if(x==f[x]) return x;
    return f[x]=find1(f[x]);
}



int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cin>>n>>m;
    for(int i=1;i<=n;i++) f[i]=i;
    for(int i=1;i<=m;i++)
    {
        int k;
        cin>>k;
        if(k==2)
        {
            cin>>a[i].u>>a[i].v>>a[i].w;
        }
        else
        {
            cin>>a[i].u>>a[i].w;
            a[i].v=a[i].u;
        }
    }
    sort(a+1,a+1+m);
    ll ans=0;
    for(int i=1;i<=m;i++)
    {
        int x=a[i].u,y=a[i].v,w=a[i].w;
        x=find1(x),y=find1(y);
        if(x!=y)
        {
            if(vis[x]&&vis[y]) continue;
            ans+=w;
            f[x]=y;
            vis[y]|=vis[x];
        }
        else
        {
            if(vis[x]) continue;
            vis[x]=1;
            ans+=w;
        }
    }
    cout<<ans<<'\n';





}