(DFS)问题 1548: [蓝桥杯][算法提高VIP]盾神与砝码称重(C++)

题目描述
有一天,他在宿舍里无意中发现了一个天平!这 个天平很奇怪,有n个完好的砝码,但是没有游码。盾神为他的发现兴奋不已!于是他准备去称一称自己的东西。他准备好了m种物品去称。神奇的是,盾神一早就 知道这m种物品的重量,他现在是想看看这个天平能不能称出这些物品出来。但是盾神稍微想了1秒钟以后就觉得这个问题太无聊了,于是就丢给了你。

数据规模和约定
1< =n< =24, 1< =m< =10.
输入
第一行为两个数,n和m。
第二行为n个数,表示这n个砝码的重量。
第三行为m个数,表示这m个物品的重量。
输出
输出m行,对于第i行,如果第i个物品能被称出,输出YES否则输出NO。

样例输入
4  2 
1  2  4  8 
15  16 
样例输出
YES
NO

本题使用DFS来解决,首先物品是放在左盘,砝码可以放在左盘也可以放在右盘。
(1)当放在左盘时物品重量加砝码重量,放在右盘时物品重量减砝码重量,当物品重量为0时说明可以称出此物品重量。
(2)这里设置了一个remain数组,表示剩余砝码的重量,如果当前物品重量大于剩余砝码重量,那么剪枝,直接返回。

#include <bits/stdc++.h>

using namespace std;
int w[25];
int remain[25]= {0};
int ok;
int n;
void DFS(int t,int f)
{
    if(t==0)
    {
        ok=1;
        return ;
    }
    if(abs(t)>remain[f])
        return ;
    for(; f<n; f++)
    {
        DFS(t-w[f],f+1);
        DFS(t+w[f],f+1);
    }
}
int main()
{
    int m;
    cin>>n>>m;
    for(int i=0; i<n; i++)
        cin>>w[i];
    sort(w,w+n);
    for(int i=n-1; i>=0; i--)
        remain[i]=w[i]+remain[i+1];
    while(m--)
    {
        ok=0;
        int thing;
        cin>>thing;
        DFS(thing,0);
        if(ok)
            cout<<"YES"<<endl;
        else
            cout<<"NO"<<endl;
    }
    return 0;
}


评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值