【笔试】给出N个物品价格，每件物品最多选一件，是否能恰好组合金额M

题目

给出N个物品价格，每件物品最多选一件，是否能恰好组合金额M。能输出1，不能输出0。
输入
输入 N 
输入 N 个价格p
输入金额 M
输出
能组合出来，输出 1，否则输出 0

样例输入
6
99 199 1999 10000 39 1499
10238
样例输出
1

 

解答

深度遍历，对于一个物品每次有两个选择：选取或不选取。注意剪枝。

#include<iostream>
#include<vector>
using namespace std;
int N, M;
vector<int> v;
bool res = false;
void dfs(int num, int m) {
    if (res) return;
    if (m == 0) {
        res = true;
        return;
    }
    if (m <0 || num >= N) return;
    else {
        dfs(num + 1, m - v[num]);
        dfs(num + 1, m);
    }
}
int main() {
    cin >> N;
    for (int i = 0; i < N; i++) {
        int tmp;
        cin >> tmp;
        v.push_back(tmp);
    }
    cin >> M;
    res = false;
    dfs(0, M);
    if (res) cout << 1;
    else cout << 0;
}