和为K的组合

最新推荐文章于 2023-03-22 10:49:55 发布

brucehb

最新推荐文章于 2023-03-22 10:49:55 发布

阅读量827

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/brucehb/article/details/69053945

版权

算法专栏收录该内容

636 篇文章 2 订阅

订阅专栏

本文介绍了一种解决子集和问题的方法，即判断一个整数数组中是否存在若干个数的和等于给定的目标值K。通过使用动态规划思想，文章详细解释了如何利用布尔型数组跟踪可能的和，并逐步检查每个输入数值以确定最终结果。

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

给出N个正整数组成的数组A，求能否从中选出若干个，使他们的和为K。如果可以，输出："Yes"，否则输出"No".

Input

第1行：2个数N, K, N为数组的长度, K为需要判断的和(2 <= N <= 20，1 <= K <= 10^9)
第2 - N + 1行：每行1个数，对应数组的元素A[i] (1 <= A[i] <= 10^6)

Output

如果可以，输出："Yes"，否则输出"No"。

Input示例

Output示例

No

#include <iostream>
#include <vector>
using namespace std;

void fun(vector<int> &input, int N, long long int K)
{
    vector<bool> flag(K+1, false);
    flag[0] = true;
  
    for (int i = 0; i < N; i++)
    {
        for (long long int j = K; j > 0; j--)
        {
            if ((!flag[j]) && (j-input[i] >= 0) && (flag[j-input[i]]))
            {
                flag[j] = true;
                if (j == K)
                {
                    cout << "Yes" << endl;
                    return;
                }
            }
        }
    }
    
    cout << "No" << endl;
}

int main()
{
    int N;
    int K;
    scanf("%d %d", &N, &K);
    
    vector<int> input;
    int a;
    for (int i = 0; i < N; i++)
    {
        scanf("%d", &a);
        input.push_back(a);
    }
    
    fun(input, N, K);
    
    return 0;
}