DFS剪枝算法经典题目-挑选

4954. 挑选 - AcWing题库

给定一个包含 n 个正整数 a1,a2,…,an的集合。

集合中可能存在数值相同的元素。

请你从集合中挑选一些元素,要求同时满足以下所有条件:

  1. 被选中元素不少于 2 个。
  2. 所有被选中元素之和不小于 l 且不大于 r。
  3. 所有被选中元素之中最大元素与最小元素之差不小于 x。

请问,一共有多少种不同的选法。

注意:

  • 不考虑元素顺序,{a1,a2} 和 {a2,a1} 应当视为同一种选法。
  • 不同元素即使数值相同,也应当视为不同个体,即使 a1=a2,{a1,a3}和 {a2,a3} 也应当视为不同选法。
输入格式

第一行包含四个整数 n,l,r,x。

第二行包含 n 个整数 a1,a2,…,an。

输出格式

一个整数,表示不同选法的数量。

数据范围

前 33 个测试点满足 1≤n≤5。
所有测试点满足 1≤n≤15,1≤l≤r≤109,1≤x≤106,1≤ai≤106。

输入样例1:
3 5 6 1
1 2 3
输出样例1:
2
输入样例2:
4 40 50 10
10 20 30 25
输出样例2:
2
输入样例3:
5 25 35 10
10 10 20 10 20
输出样例3:
6

根据题目,因为不考虑选取的顺序,所以可以根据下标进行dfs,类似根据下标求组合数。

 首先遍历数组,从每一个数开始dfs,参数分别为当前搜索的数的总和,搜索到的数组的下标,搜索数中最大值,搜索数种最小值,搜索数的个数,然后每次dfs的时候,判断当前总数是否大于 r ,如果大于则不用继续搜索了,可以直接剪掉。然后通过参数判断是否符合条件,如果符合则ans++,然后继续遍历,从index+1开始遍历,更新参数继续dfs,最终结果就为ans。

AC code:

#include<bits/stdc++.h>
using namespace std;
int n, l, r, x;
int arr[20];
int ans = 0;
void dfs(int total, int index, int mx, int mi, int cnt) {
	if (total > r ) return ;
	if (total >= l && total <= r && (mx - mi) >= x && cnt > 1) ans++;

	for (int i = index + 1; i <= n; i++) {
		int a = max(mx, arr[i]);
		int b = min(mi, arr[i]);
		dfs(total + arr[i], i, a, b, cnt + 1);
	}
}
int main() {
	cin >> n >> l >> r >> x;
	for (int i = 1; i <= n; i++) cin >> arr[i];
	for (int i = 1; i <= n; i++) {
		dfs(arr[i], i, arr[i], arr[i], 1);
	}
	cout << ans;
}

相关推荐

  1. DFS剪枝算法经典题目-挑选

    2024-02-03 00:10:01       55 阅读
  2. 167. 木棒(dfs剪枝经典题)

    2024-02-03 00:10:01       62 阅读
  3. <span style='color:red;'>dfs</span><span style='color:red;'>剪枝</span>

    dfs剪枝

    2024-02-03 00:10:01      45 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-02-03 00:10:01       98 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-02-03 00:10:01       106 阅读
  3. 在Django里面运行非项目文件

    2024-02-03 00:10:01       87 阅读
  4. Python语言-面向对象

    2024-02-03 00:10:01       96 阅读

热门阅读

  1. 把 matlab 公式输出成 latex 公式形式

    2024-02-03 00:10:01       49 阅读
  2. python怎样把形似列表的字符串转换为列表?

    2024-02-03 00:10:01       52 阅读
  3. uniapp生命应用生命周期和页面生命周期

    2024-02-03 00:10:01       53 阅读
  4. LeetCode 面试热门 100 题(第 038 ~ 050题)

    2024-02-03 00:10:01       56 阅读
  5. 常用的数据库SQL语句使用大全

    2024-02-03 00:10:01       48 阅读
  6. 面了搜狐大模型算法岗(实习),有惊无险。。。

    2024-02-03 00:10:01       48 阅读
  7. 网络编程练习题(TCP)

    2024-02-03 00:10:01       49 阅读
  8. 一线大厂面试真题——谈谈你对ES的理解

    2024-02-03 00:10:01       52 阅读
  9. linux 上库五步走,经验贴

    2024-02-03 00:10:01       50 阅读