深搜树模型

另一个人。

已于 2023-07-30 13:57:52 修改

阅读量71

点赞数 3

分类专栏：数据结构与算法文章标签：算法数据结构

于 2023-02-25 10:22:51 首次发布

本文链接：https://blog.csdn.net/m0_73604812/article/details/129212562

版权

数据结构与算法专栏收录该内容

22 篇文章 0 订阅

订阅专栏

该程序使用深度优先搜索策略解决从n个数中选取k个数，使其和等于s的问题。通过递归的dfs函数，尝试选择和不选择当前元素，同时检查是否满足和为s的条件。如果找到解，则设置标志位flag为true，最终输出YES或NO表示是否存在这样的子集。

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


#include<bits/stdc++.h>
using namespace std;
//深搜树模型
/*n个数  从中选k个数  使k个数之和为s
若存在输出YES，否则输出NO
input:
n:3 k:2 s:6
arr:1 2 5
output:
YES
*/
int n, k, s;
int a[101];
bool flag = false;
void dfs(int m, int sum, int dep)
{
    if (m == k && sum == s)
    {
        flag = true;
        return;
    }
    if (sum >= s) return;
    if (dep == n) return;
    dfs(m + 1, sum + a[dep], dep + 1);  //选
    if (flag) return;//最优性剪枝
    dfs(m, sum, dep + 1);  //不选
}
int main()
{
    cin >> n >> k >> s;
    for (int i = 0; i < n; i++)
        cin >> a[i];
    dfs(0, 0, 0);  //初始状态
    cout << (flag ? "YES" : "NO");
    return 0;
}