牛客寒假训练 魏迟燕的自走棋 贪心+并查集

题目在这
贪心方案：对于装备的价值从大到小进行选择。

如果每件装备只对应一个人的话，只需要设置一个st数组记录每个人知否已经选了装备，从大到小枚举每件装备能否选择即可。

现在问题是k=1或k=2，我们可以考虑将士兵看成点，用并查集来维护士兵的信息，使用st数组来维护连通块能否还能选择装备。
连通块的意义：对于一个连通块，它可能不能再匹配装备了，或者还能匹配一件装备。如果还能匹配一件装备的话，任意选择其中一个点，则之前匹配该连通块的装备能匹配剩下的全部点。因此只要有装备能匹配连通块中的任意点，都可以将该装备选择。

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;
typedef long long LL;
const int N = 1e5 + 10;
const int M = 2 * N;
int n, m, st[N], fa[N];
struct Node
{
    int w, a, b;
    bool operator < (const Node& m)
    {
        return w > m.w;
    }
}thg[N];

int get(int x)
{
    if (fa[x] == x) return x;
    return fa[x] = get(fa[x]);
}

int main()
{
    LL res = 0;
    cin >> n >> m;
    for (int i = 1; i < N; i ++)   fa[i] = i;
    for (int i = 0; i < m; i ++)
    {
        int k; cin >> k;
        if (k == 1)
            cin >> thg[i].a >> thg[i].w;
        else
            cin >> thg[i].a >> thg[i].b >> thg[i].w;
    }
    sort(thg, thg + m);
    
    for (int i = 0; i < m; i ++)
    {
        int a = thg[i].a, b = thg[i].b, w = thg[i].w;
        if (b)
        {
            int pa = get(a), pb = get(b);
            if (pa == pb)
            {
                if (!st[pa])
                {
                    res += w;
                    st[pa] = 1;
                }
            }
            else
            {
                if (!st[pa] || !st[pb])
                {
                    fa[pa] = pb;
                    st[pb] = !(!st[pa] && !st[pb]);
                    res += w;
                }
            }
        }
        else
        {
            int pa = get(a);
            if (!st[pa])
            {
                st[pa] = 1;
                res += w;
            }
        }
    }
    cout << res;
    return 0;
}