01背包
题干描述
有n个重量和价值分别为wi,vi。从这些物品中挑选出总重量不超过w的物品,求所挑选方案中价值总和的最大值。
限制条件
- 1 ≤ \leq ≤ n ≤ \leq ≤ 100
- 1 ≤ \leq ≤ wi,vi ≤ \leq ≤ 100
- 1 ≤ \leq ≤ W ≤ \leq ≤ 10000
1.搜索解法
这道题最朴素的方法就是采用递归对于每一个物品都考虑一下选或者不选进行搜索试试看。
暴力代码
int rec(int i,int j)
{
int res;
if(i==n){
//已经没有剩余物品了
res=0;
}else if(j<w[i]){
//无法挑选这个物品
res=rec(i+1,j);
}else{
//挑选和不挑选的两种情况都尝试一下
res=max(rec(i+1,j),res(i+1,j-w[i])+v[i]);
}
//将结果记录在数组中
return res;
}
void solve(){
printf("%d\n",rec(0,W));
}
记忆搜索
int dp[Max_N+1][Max_W+1];//记忆化数组
int rec(int i,int j)
{
if(d[i][j]>=0){
//已经计算过的话直接使用之前的结果
return dp[i][j];
}
int res;
if(i==n){
//已经没有剩余物品了
res=0;
}else if(j<w[i]){
//无法挑选这个物品
res=rec(i+1,j);
}else{
//挑选和不挑选的两种情况都尝试一下
res=max(rec(i+1,j),res(i+1,j-w[i])+v[i]);
}
//将结果记录在数组中
return dp[i][j]=res;
}
void solve(){
memset(dp,-1,sizeof(dp));
printf("%d\n",rec(0,W));
}
缺点
暴力代码的搜索深度是n,而且每一层都需要两个分支,所有最坏的情况下的复杂度就是o(2
n
^n
n),如果n一旦比较大了之后就会TLE。优化代码采用了一个记忆化的数组,当dp[i][j]不为零的时候就直接返回对应的值便可以。但是如果Max_N和Max_W
≥
\geq
≥ 1e5 之后空间复杂度就会升高,在限定内存的情况下,dp[i][j]很可能也会发生MLE的情况。
2.动态规划
dp[i+1][j]:=从0到i这i+1个物品中选出总重量不超过j的物品时总价值的最大值。
转移方程
d
p
[
i
+
1
]
[
j
]
=
{
d
p
[
i
]
[
j
]
,
(
j
<
w
[
i
]
)
m
a
x
(
d
p
[
i
]
[
j
]
,
d
p
[
i
]
[
j
−
w
[
i
]
]
+
v
[
i
]
)
(
其
他
)
dp[i+1][j]= \begin{cases} dp[i][j],&(j<w[i])\\ max(dp[i][j],dp[i][j-w[i]]+v[i])&(其他) \end{cases}
dp[i+1][j]={dp[i][j],max(dp[i][j],dp[i][j−w[i]]+v[i])(j<w[i])(其他)
int dp[Max_N+1][Max_W+1];
void solve()
{
for(int i=0;i<n;i++)
{
for(int j=0;j<=W;j++)
{
if(j<w[i])
dp[i+1][j]=dp[i][j];
else
dp[i+1][j]=max(dp[i][j],dp[i][j-w[i]]+v[i]);
}
}
printf("%d\n",dp[n][W]);
}
但是此方法在洛谷P2871这道题中Max_N为3405,Max_W为12855,内存限制为128MB,会出现MLE类型的错误
空间优化
为了解决上文所出现的问题,那么我们就要考虑进行空间上的优化,仔细观察转移方程后会发现dp[i+1][j]只会用到上一行的dp[i][j]或者dp[i][j-w[i]]+v[i]。
那么我们就可以考虑用只一个一维数组不断的更新,省去二维数组中的行变量。
在第i次循环之前dp[j]表示第i-1次循环之后的结果,即为dp[j]=dp[i-1][j]。
在第i次循环之后dp[j]表示dp[i][j]。
那么转移方程改变为
d
p
[
j
]
=
{
d
p
[
j
]
(
j
<
w
[
i
)
d
p
[
j
−
w
[
i
]
]
+
v
[
i
]
(
其
他
)
dp[j]= \begin{cases} dp[j]&(j<w[i)\\ dp[j-w[i]]+v[i]&(其他) \end{cases}
dp[j]={dp[j]dp[j−w[i]]+v[i](j<w[i)(其他)
并且在此过程中我们也要时刻保证转移方程中dp[j]表示的是dp[i-1][j],dp[j-w[i]]表示的是dp[i-1][j-w[i]]。若j变量为从小到大,那么在dp[j-w[i]]已经更新,表示的是dp[i][j-w[i]]而不是dp[i-1][j-w[i]]。
代码
void solve()
{
for(int i=0;i<n;i++)
{
for(int j=W;j>=w[i];j--)
{
dp[j]=max(dp[j],dp[j-w[i]]);
}
}
printf("%d\n",dp[W]);
}