北京大学考研机试题——神奇的口袋

题目描述

有一个神奇的口袋，总的容积是 $40$，用这个口袋可以变出一些物品，这些物品的总体积必须是 $40$。

John 现在有 $n$ 个想要得到的物品，每个物品的体积分别是 $a1,a2\dots \dots an$。

John 可以从这些物品中选择一些，如果选出的物体的总体积是 $40$，那么利用这个神奇的口袋，John 就可以得到这些物品。

现在的问题是，John 有多少种不同的选择物品的方式。

输入格式

输入的第一行是正整数 $n$，表示不同的物品的数目。

接下来的 $n$ 行，每行有一个 $1$ 到 $40$ 之间的正整数，分别给出 $a1,a2\dots \dots an$ 的值。

输出格式

输出不同的选择物品的方式的数目。

数据范围

$1\le n\le 20$,
$1\le ai\le 40$

输入样例：

3
20
20
20

输出样例：

3

此题与01背包较为类似.优化方案也几乎一致，我们每次只用到了上一次的值i-1因此可以可以优化掉二维

代码如下

#include<iostream>
using namespace std;

int f[50][50];
int n,w[50];

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>w[i];
    f[0][0]=1;//初始化
    for(int i=1;i<=n;i++)
        for(int j=0;j<=40;j++)
        {
            f[i][j]=f[i-1][j];
            if(j>=w[i]) f[i][j]+=f[i-1][j-w[i]];
        }
    cout<<f[n][40]<<endl;
    return 0;
}

---

优化版本

#include<iostream>
using namespace std;

int f[50];
int n,w[50];

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>w[i];
    f[0]=1;//初始化
    for(int i=1;i<=n;i++)
        for(int j=40;j>=w[i];j--)//从大到小 从小到大会先使用f[40]的值与二维版本不符
        {
            f[j]+=f[j-w[i]];
        }
    cout<<f[40]<<endl;
    return 0;
}