[牛客网题库] 牛牛的背包问题

牛牛准备参加学校组织的春游, 出发前牛牛准备往背包里装入一些零食, 牛牛的背包容量为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

输出例子1:

8

例子说明1:

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

时空限制：

时间限制：1秒
空间限制：32768K

解题思路：
1、先计算总体积，如果总体积小于背包容量，则返回2^n（零食总类）
2、如果总体积大于背包容量，那么每一种零食都有取和不取两种方法，是组合问题。

#include <iostream>
#include <cstdio>
#include <stdlib.h>
#include <math.h>
#include <algorithm>
using namespace std;

class Solution{
public:
    int kinds(int foods[], int n, int w){
        if(w == 0) return 0;

        int res = 0;
        sort(foods, foods + n);
        int totalW = 0;
        bool isFull = false;    // 和溢出判断
        for(int i = 0; i < n; ++i){
            totalW += foods[i];
            if(totalW < 0){    // 如果溢出
                isFull = true;
                break;
            }
        }
        if(!isFull && totalW < w)    // 没有溢出并且背包容量大于总食品重量
            return pow(2, n);

        backtracing(foods, n, w, 0, res);

        return res ;
    }

private:
    void backtracing(int foods[], int n, int pack, int begin, int &res){
        if(begin == n || foods[begin] > pack){
            res++;
            return;
        }

        pack -= foods[begin];  
        backtracing(foods, n, pack, begin + 1, res);
        pack += foods[begin];
        backtracing(foods, n, pack, begin + 1, res);
    }
};

int main(){
    int n, w;
    cin >> n >> w;
    int foods[n];
    for(int i = 0; i < n; ++i){
        cin >> foods[i];
    }
    Solution s;
    std:cout << s.kinds(foods, n, w) << std::endl;
    return 0;
}