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