上海计算机学会 2023年9月月赛乙组T1 子集和（七）（折半搜索、二分）

最新推荐文章于 2024-08-27 17:53:55 发布

超哥聊信奥

最新推荐文章于 2024-08-27 17:53:55 发布

阅读量744

点赞数 28

分类专栏：题目题解文章标签：算法 c++ 数据结构图论学习

本文链接：https://blog.csdn.net/oitutor/article/details/137497068

版权

题目题解专栏收录该内容

96 篇文章 1 订阅

订阅专栏

本文介绍了一种在给定范围内计算满足条件子集和的方法，利用折半搜索和排序策略，有效降低时间复杂度，适用于IT竞赛或编程挑战。

摘要由CSDN通过智能技术生成

第一题：T1子集和（七）

标签：折半搜索、二分
题意：给定 $n$ 个数字 $a_1,a_2,…,a_n$ 构成一个集合（本题中所指的集合元素允许相等），请计算有多少子集的和大于 $0$ ？（ $1 \leq n \leq 40$ ， $10^9≤a_i≤10^9$ ）
题解： $n$ 的数据范围是 $40$ ，直接暴力搜索时间复杂度会到达 $2^{40}$ ，肯定会超时，所以得先分成两个集合，折半搜索处理一下，然后对两个集合进行排序。对第一个集合进行遍历，通过二分，去第二个集合中找到第一个大于 $- s 1 [i]$ 的位置 $j$ ，因为已经排序过了，从 $j$ 到第二个集合最后一个元素都是能够加起来和大于 $0$ 的。
代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const ll N = 2e6;
ll n, a[50], ans = 0;
ll s1[N], s2[N], c1 = 0, c2 = 0;

void dfs1(ll p, ll sum) {
    if (p == n / 2 + 1) {
        s1[c1++] = sum;
        return ;
    }
    dfs1(p + 1, sum + a[p]);
    dfs1(p + 1, sum);
}

void dfs2(ll p, ll sum) {
    if (p == n + 1) {
        s2[c2++] = sum;
        return ;
    }
    dfs2(p + 1, sum + a[p]);
    dfs2(p + 1, sum);
}

int main() {
    cin >> n;
    for (int i = 1; i <= n; i++) cin >> a[i];
    dfs1(1, 0);
    dfs2(n / 2 + 1, 0);
    sort(s1, s1 + c1);
    sort(s2, s2 + c2);

    for (int i = 0; i < c1; i++) {
        ll j = upper_bound(s2, s2 + c2, -s1[i]) - s2;
        ans += c2 - j;
    }

    cout << ans;
    return 0;
}