【题目链接】
ybt 1291:数字组合
OpenJudge NOI 2.6 2985:数字组合
【题目考点】
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
∗
n
)
O(m*n)
O(m∗n),空间复杂度为
O
(
m
∗
n
)
O(m*n)
O(m∗n)。
可以进行滚动数组优化,定义状态时使用一维数组而不是二维数组,时间复杂度为
O
(
m
∗
n
)
O(m*n)
O(m∗n),空间复杂度为
O
(
m
)
O(m)
O(m)
【解题思路】
根据本题题意,每个数字应该只能使用一次,那么这题就可以归类为01背包问题的范畴。
本题为:01背包求填满背包的方案数问题。
该题中,给定的n个整数对应每件物品的重量,t对应背包大小。用上述方法求解即可。
【题解代码】
解法1:二维状态
#include<bits/stdc++.h>
using namespace std;
#define N 25
#define T 1005
int dp[N][T], a[N];//a[i]:第i个数 dp[i][j]:前i个数中选择一些数字加和为j的方案数
int n, t;
int main()
{
cin >> n >> t;
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 <= t; ++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][t];
return 0;
}
解法2:滚动数组优化 一维状态
#include<bits/stdc++.h>
using namespace std;
#define N 25
#define T 1005
int dp[T], a[N];//a[i]:第i个数 dp[i][j]:前i个数中选择一些数字加和为j的方案数
int n, t;
int main()
{
cin >> n >> t;
for(int i = 1; i <= n; ++i)
cin >> a[i];
dp[0] = 1;//初始状态:前0个数字中选择数字加和加和凑数字0,有1种方案
for(int i = 1; i <= n; ++i)
for(int j = t; j >= a[i]; --j)
dp[j] = dp[j] + dp[j-a[i]];
cout << dp[t];
return 0;
}