子集和的元素和【二分】【DFS】

最新推荐文章于 2024-09-19 22:15:00 发布

ssl_fuyang

最新推荐文章于 2024-09-19 22:15:00 发布

阅读量138

点赞数

分类专栏：二分 DFS;BFS 文章标签：深度优先算法 c++

本文链接：https://blog.csdn.net/qq_43010386/article/details/126677474

版权

DFS;BFS 同时被 2 个专栏收录

52 篇文章 0 订阅

订阅专栏

二分

31 篇文章 0 订阅

订阅专栏

本文介绍了如何使用二分搜索配合深度优先算法，并结合剪枝技巧，来解决寻找元素和小于给定值k的子集数量的问题。通过将集合A按降序排列，可以显著提高算法效率，达到题目要求的60分解决方案。

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

>Link

SSLOJ

>Description
在这里插入图片描述
$1≤n≤50000, 1≤k≤200000，1≤Ai≤10^9$

>解题思路

这道题竟然二分+爆搜+剪枝就能过
二分答案，爆搜来找和比 $k$ 小的集合有多少个，加一些常规的剪枝
但是这样是60分，还有一个重要的剪枝就是把集合 $A$ 从大到小排序，然后再爆搜，这样就可以跑得飞快，因为减少了很多不必要的搜索

>代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 50010
#define LL long long
using namespace std;

int n, k;
LL a[N], cnt, cntt, ans;

LL read ()
{
	LL ret = 0;
	char c = getchar();
	while (c < '0' || c > '9') c = getchar();
	while (c >= '0' && c <= '9')
	{
		ret = ret * (LL)10 + c - '0';
		c = getchar();
	}
	return ret;
}
void dfs (int now, LL sum, LL w)
{
	if (sum > w) return;
	if (cnt >= k) return;
	if (now == 0)
	{
		if (sum < w && sum != 0) cnt++;
		if (sum == w) cntt++;
		return; 
	}
	dfs (now - 1, sum, w);
	dfs (now - 1, sum + a[now], w);
}

int main()
{
//	freopen ("16.in", "r", stdin);
	n = read(), k = read();
	LL l = 1, r = 0, mid;
	for (int i = 1; i <= n; i++)
	{
		a[i] = read();
		r += a[i];
	}
	sort (a + 1, a + 1 + n);
	while (l <= r)
	{
		mid = (l + r) / 2;
		cnt = 0;
		cntt = 0;
		dfs (n, 0, mid);
		if (cnt < k && cnt + cntt >= k)
		{
			ans = mid;
			break;
		}
		if (cnt >= k) r = mid - 1;
		else l = mid + 1;
	}
	printf ("%lld", ans);
	return 0;
}