美文网首页
零食放法

零食放法

作者: 王王王王王景 | 来源:发表于2019-08-18 20:22 被阅读0次

    牛牛准备参加学校组织的春游, 出发前牛牛准备往背包里装入一些零食, 牛牛的背包容量为w。

    牛牛家里一共有n袋零食, 第i袋零食体积为v[i]。

    牛牛想知道在总体积不超过背包容量的情况下,他一共有多少种零食放法(总体积为0也算一种放法)。

    输入描述:

    输入包括两行
    第一行为两个正整数n和w(1 <= n <= 30, 1 <= w <= 2 * 10^9),表示零食的数量和背包的容量。
    第二行n个正整数v[i](0 <= v[i] <= 10^9),表示每袋零食的体积。
    

    输出描述:

    输出一个正整数, 表示牛牛一共有多少种零食放法。
    

    示例1

    输入

    3 10
    1 2 4
    

    输出

    8
    

    说明

    三种零食总体积小于10,于是每种零食有放入和不放入两种情况,一共有2 * 2 * 2 = 8种情况。

    有条件的子集问题:
    特殊情况可以快速求解

    #include <iostream>
    #include <vector>
    #include <algorithm>
    #include <math.h>
    using namespace std;
    
    void func(int &re, long sum, const long w, const vector<long> &vec, int k) {
        if(sum > w) {
            return;
        }
        ++re;
        for(int i = k; i < vec.size(); ++i)
            func(re, sum + vec[i], w, vec,  i + 1);
    }
    
    int main()
    {
        int n;
        long w;
        cin>>n>>w;
        vector<long> vec;
        long all = 0;
        for(int i = 0; i < n; ++i) {
            long temp= 0;
            cin>>temp;
            vec.push_back(temp);
            all += temp;
        }
        int re = 0;
        if(all <= w)    
            re = pow(2, n);
        else {
            // sort(vec.begin(), vec.end());
            func(re, 0, w, vec, 0);
        }
        
        cout<<re<<endl;
        return 0;
    }
    

    相关文章

      网友评论

          本文标题:零食放法

          本文链接:https://www.haomeiwen.com/subject/ousdsctx.html