[蓝桥杯][算法提高]ADV-192 盾神与砝码称重 【DFS+剪枝】 题解和C++示例代码

ADV-192 算法提高 盾神与砝码称重

资源限制
时间限制：1.0s 内存限制：256.0MB

问题描述
　　有一天，他在宿舍里无意中发现了一个天平！这个天平很奇怪，有n个完好的砝码，但是没有游码。盾神为他的发现兴奋不已！于是他准备去称一称自己的东西。他准备好了m种物品去称。神奇的是，盾神一早就知道这m种物品的重量，他现在是想看看这个天平能不能称出这些物品出来。但是盾神稍微想了1秒钟以后就觉得这个问题太无聊了，于是就丢给了你。
　　
输入格式
　　第一行为两个数，n和m。
　　第二行为n个数，表示这n个砝码的重量。
　　第三行为m个数，表示这m个物品的重量。
　　
输出格式
　　输出m行，对于第i行，如果第i个物品能被称出，输出YES否则输出NO。
　　
样例输入
4 2
1 2 4 8
15 16

样例输出
YES
NO

样例输入
4 1
10 7 1 19
6

样例输出
YES

数据规模和约定
　　1<=n<=24, 1<=m<=10.

题解：

DFS:

dfs列举每种状态：
每个砝码 3 种状态，假设左边为正，右边为负，遍历 3 种状态即可。

• dfs(num - 1, sum + datas[num]); 第 num 个砝码放左边。
• dfs(num - 1, sum); 这个砝码不放。
• dfs(num - 1, sum - datas[num]); 第 num 个砝码放右边。

60分 dfs+剪枝:

#include<iostream>
using namespace std;
int weight[25];
int flag;
int n, m;
int thing;

void dfs(int sum, int num) {

    if (sum == thing) {
        flag = 1;
        return;
    }
    if (num < 0) {
        return;
    }
    dfs(sum + weight[num], num - 1);
    dfs(sum, num - 1);
    dfs(sum - weight[num], num - 1);
}

int main() {
    cin >> n >> m;
    for (int i = 0; i < n; i++) {
        cin >> weight[i];
    }

    for (int i = 0; i < m; i++) {
        cin >> thing;
        flag = 0;
        dfs(0, n - 1);
        if (flag == 1) {
            cout << "YES" << endl;
        }
        else {
            cout << "NO" << endl;
        }
    }

    return 0;
}

100分 dfs+剪枝:

#include<iostream>
#include<algorithm>
using namespace std;
int weight[25], k[25];
int flag;
int n, m;
int thing;//要称的东西质量 

void dfs(int sum, int num) {
    if (flag == 1) {//剪枝 
        return;
    }
    if (num < 0) {
        return;
    }
    if (abs(thing - sum) > k[num]) {//如果物品的重量大于剩下的砝码能表示的最大重量，无法称出 
        return;
    }

    if (sum == thing || abs(thing - sum) == k[num]) {
        flag = 1;
        return;//退出条件 
    }

    dfs(sum + weight[num], num - 1);
    dfs(sum, num - 1);
    dfs(sum - weight[num], num - 1);
}

int main() {
    scanf("%d%d", &n, &m);
    for (int i = 0; i < n; i++) {
        scanf("%d", &weight[i]);
    }
    sort(weight, weight + n); //砝码从小到大排序 

    k[0] = weight[0];
    for (int i = 1; i < n; i++) {  //依次求出前 i个砝码能称出的最大重量
        k[i] = weight[i] + k[i - 1];//1,3,7,15
    }

    for (int i = 0; i < m; i++) {
        scanf("%d", &thing);
        flag = 0;
        dfs(0, n - 1);
        if (flag == 1) 
            printf("YES\n");
        else 
            printf("NO\n");
    }

    return 0;
}