计蒜课 Miku and Generals （并查集+01背包 dp）

题目链接：https://nanti.jisuanke.com/t/39271

题意：

n个人，每个人有一个值，分两堆。m个关系，A和B一定不能在一堆。把所有人分两堆，输出堆值（堆值=每个堆中所有人值的和）相差最小的情况下，较大的堆值。

思路：

有m个关系，利用并查集可以组成若干个（num个）联通块，每一块又可以分成两小块，每个小块中的人一定属于同一堆，不同小块属于不同堆，计算出每一小堆的值，就把问题转化成了把2*num个值分两堆，使两堆差值最小（见Ural 1005）。考虑dp，问题进一步转化为n个物品取若干，其重量和不超过sum/2（sum是所有值的总和），且重量达到最大，01背包问题。

代码：

#include<iostream>
#include<map>
#include<algorithm>
#include<vector>
#include<cstring>
#include<cstdio>
using namespace std;
const int maxn = 1e5+5;

map<int, int>mp;
int t, n, m, fa[maxn], val[maxn], a[maxn], dp[maxn];
vector<int> bag[maxn][2], vec;
int Find(int x){
    if(fa[x] == x) return x;
    int temp = fa[x];
    fa[x] = Find(fa[x]);
    val[x] = (val[x] + val[temp]) % 2;
    return fa[x];
}
void Merge(int x, int y, int w){
    int xx = Find(x), yy = Find(y);
    if (xx == yy) return ;
    fa[yy] = xx;
    val[yy] = (val[x] + val[y] + w ) % 2;
}
int main()
{
    scanf("%d", &t);
    while(t--){
        mp.clear();
        vec.clear();
        memset(dp, 0, sizeof(dp));
        for(int i=1; i<=n; i++ ) bag[i][0].clear(), bag[i][1].clear();
        memset(bag, 0, sizeof(bag));
        scanf("%d%d", &n, &m);
        for(int i=1; i<=n; i++) fa[i] = i, val[i] = 0;
        for(int i=1; i<=n; i++) {scanf("%d", &a[i]); a[i] /= 100;}
        for(int i=1; i<=m; i++) {
            int u, v;
            scanf("%d%d", &u, &v);
            Merge(u, v, 1);
        }
        int num = 0;
        for(int i=1; i<=n; i++) {    //num个联通块，每一个分成两小块，bag[num][0]和bag[num][1]
            int tmp = Find(i);
            if(mp.find(tmp) == mp.end()) mp[tmp] = ++num;
            bag[mp[tmp]][val[i]].push_back(i);
        }
       
        //计算每一小堆的值，计算完成后任务转化为把2*num个值分两堆，使两堆之差最小
        for(int i=1; i<=num; i++){
            int sum = 0;
            for(auto j : bag[i][1])
                sum += a[j];
            vec.push_back(sum);

            sum = 0;
            for(auto j : bag[i][0])
                sum += a[j];
            vec.push_back(sum);
        }

        int sum = 0;
        for(auto i : vec) sum += i;
        for(auto i : vec)
            for(int k = sum; k>=i; k--)
                dp[k] = max(dp[k], dp[k-i] + i);
            
        int ans, minn = 0x7fffffff;
        for(int i=0; i<=sum; i++){
            if(dp[i] && abs(sum-dp[i]-dp[i]) <= minn){ //sum-dp[i]为一堆,dp[i]为另一堆
                minn = abs(sum-dp[i]-dp[i]);
                ans = max(sum-dp[i], dp[i]);
            }
        }
        printf("%d\n", ans*100);
    }
}