【poj 3977】Subset (折半枚举)

最新推荐文章于 2023-09-23 07:35:39 发布

Just do it

最新推荐文章于 2023-09-23 07:35:39 发布

阅读量93

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/hs-zlq/p/11230696.html

本文分享了一段使用C++实现的算法竞赛代码，通过计算数组中两部分绝对值最小和来寻找最优划分，涉及数组操作、二进制位运算、排序及搜索等技巧。适用于算法竞赛选手和C++程序员学习参考。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
#define ll long long
#define P pair<ll, int>

ll a[40];
P p[1 << 20];

ll Abs(ll x) {
    if (x >= 0) return x;
    return -x;
}

int main() {
    int n;
    while (scanf("%d", &n) != EOF && n) {
        for (int i = 0; i < n; i++) scanf("%lld", &a[i]);
        int rn = n / 2;
        P res;
        res = {Abs(a[0]), 1};
        for (int i = 1; i < 1 << rn; i++) {
            p[i] = {0, 0};
            for (int j = 0; j < rn; j++) {
                if (i >> j & 1) {
                    p[i].first += a[j];
                    p[i].second++;
                }
            }
            //只在p[]中取数
            res = min(res, P(Abs(p[i].first), p[i].second));
        }

        sort(p + 1, p + (1 << rn));
        //去重
        int m = 2;
        for (int i = 2; i < 1 << rn; i++) {
            if (p[m - 1].first < p[i].first) p[m++] = p[i];
        }

        for (int i = 1; i < 1 << (n - rn); i++) {
            ll tmp = 0;
            int tnum = 0;
            for (int j = 0; j < n - rn; j++) {
                if (i >> j & 1) {
                    tmp += a[j + rn];
                    tnum++;
                }
            }
            //不在p[]取数
            res = min(res, P(Abs(tmp), tnum));
            int pos = lower_bound(p + 1, p + m, P(-tmp, 0)) - p;
            for (int j = -1; j <= 0; j++) {
                if (j + pos < 1 || j + pos >= m) continue;
                res = min(res, P(Abs(tmp + p[pos + j].first),
                                 tnum + p[pos + j].second));
            }
        }
        printf("%lld %d\n", res.first, res.second);
    }
}