洛谷 P1164 小A点菜

【题目链接】

洛谷 P1164 小A点菜

【题目考点】

1. 动态规划：背包问题

2. 01背包：求填满背包的方案数

问题定义

给定n件物品的重量，以及一个空间为m的背包，在n件物品中选择一些物品放入背包将背包填满，没有剩余空间。求能将背包填满的方案数量。

1. 状态定义

• 集合：将n件物品放入空间为m的背包的方案
• 限制：物品范围，背包空间
• 属性：总重量
• 条件：为m
• 统计量：方案数

状态定义：dp[i][j]表示在前i件物品中选择物品填满j的背包的方案总数。
初始状态：dp[i][0]前i个物品中选择物品，填满空间大小为0的背包，不放物品即可，这也是一种方案。因此dp[i][0]=1。注意，包括dp[0][0]=1。

2. 状态转移方程

记w[i]为第i个物品的重量
分割集合：以第i物品是否放入背包分割集合

• 子集1：如果不将第i物品放入背包，那么在前i件物品中选择物品填满大小为j的背包的方案数，即为在前i-1件物品中选择物品填满大小为j的背包的方案数，即dp[i-1][j]
• 子集2：如果第i物品可以放入背包（即j >= w[i]），且确定将第i物品放入背包，此时背包剩下空间为j-w[i]。那么在前i件物品中选择物品填满大小为j的背包的方案数，即为在前i-1件物品中选择物品放入空间为j-w[i]的背包的方案数，即dp[i-1][j-w[i]]
• 以上两种情况得到的方案都是可行的方案，因此将以上两类方案加和，即为在前i件物品中选择物品填满大小为j的背包的方案数。即dp[i][j] = dp[i-1][j] + dp[i-1][j-w[i]]

该解法为将状态定义为二维数组，时间复杂度为$O(m\ast n)$，空间复杂度为$O(m\ast n)$。
可以进行滚动数组优化，定义状态时使用一维数组而不是二维数组，时间复杂度为$O(m\ast n)$，空间复杂度为$O(m)$

【解题思路】

根据本题题意，每种菜只能点一次，那么这题就可以归类为01背包问题的范畴。
本题为：01背包求填满背包的方案数问题。
该题中，给定的N种菜的价格对应每件物品的重量，总钱数M元对应背包大小。用上述方法求解即可。

【题解代码】

解法1：二维状态

#include<bits/stdc++.h>
using namespace std;
#define N 105
#define M 10005
int dp[N][M], a[N];//a[i]:第i道菜的价钱 dp[i][j]：前i个菜中选择一些菜的价钱加和为j的方案数 
int n, m;
int main()
{
    cin >> n >> m;
    for(int i = 1; i <= n; ++i)
         cin >> a[i];
    for(int i = 0; i <= n; ++i)//设初始状态 
        dp[i][0] = 1;//前i道菜中选择一些菜花0元，不选即可，有1种方案。 
    for(int i = 1; i <= n; ++i)
        for(int j = 1; j <= m; ++j)
        {
            if(j >= a[i])
                dp[i][j] = dp[i-1][j] + dp[i-1][j-a[i]];
            else 
                dp[i][j] = dp[i-1][j];
        }
    cout << dp[n][m];
    return 0;
}

解法2：滚动数组优化 一维状态

#include<bits/stdc++.h>
using namespace std;
#define N 105
#define M 10005
int dp[M], a[N];//a[i]:第i道菜的价钱 dp[i][j]：前i个菜中选择一些菜的价钱加和为j的方案数 
int n, m;
int main()
{
    cin >> n >> m;
    for(int i = 1; i <= n; ++i)
         cin >> a[i];
    dp[0] = 1;//前i道菜中选择一些菜花0元，不选即可，有1种方案。 
    for(int i = 1; i <= n; ++i)
        for(int j = m; j >= a[i]; --j)
            dp[j] = dp[j] + dp[j-a[i]];
    cout << dp[m];
    return 0;
}