dp专题、经典题目
01背包问题
题目
思路
每个物品有拿与不拿两种状态(拿的时候还需要判断一下能不能装得下)。考虑第i
个物品时,前i-1
个物品的最大价值,相对于前i
个物品的最大价值,属于重叠子问题,这样就可以想到前i
个物品的最大价值由前i-1
个物品的最大价值得来:
- 当不拿第
i
个物品时,f[i] = f[i-1] - 当能拿第
i
个物品时,就需要考虑,单单一个状态没法表示所有的子集状态:前多少物品、体积多少的最大价值;这时就要多加一个状态f[i][j] = f[i-1][j - v[i]] + w[i];表示前i-1
个物品且体积最多为j-v[i]
时的最大价值加上第i
个物品的价值w[i]
- 毕竟两种情况,取大的值就是前
i
个物品的最大价值
代码
记忆递归
由于0<N,V≤1000
,故递归不会爆栈,这个数据规模可以AC
#include <bits/stdc++.h>
using namespace std;
const int N = 1010;
int f[N][N], v[N], w[N];
int dfs(int i, int j){
if(f[i][j] != -1) return f[i][j];
if(i == 0) return 0;
int res = dfs(i-1, j);
if(v[i] <= j) res = max(res, dfs(i-1, j - v[i]) + w[i]);
return f[i][j] = res;
}
int main(){
int n, m;
cin >> n >> m;
for(int i = 1;i<=n;i++) cin >> v[i] >> w[i];
for(int i =1;i<=n;i++)
for(int j = 0;j<=m;j++) f[i][j] = -1;
cout<< dfs(n, m) << endl;
return 0;
}
dp
假如数据规模更大,会爆栈时,就需要将记忆递归修改为循环实现
#include <bits/stdc++.h>
using namespace std;
const int N = 1010;
int f[N][N], v[N], w[N];
int main(){
int n, m;
cin >> n >> m;
for(int i = 1;i<=n;i++) {
cin >> v[i] >> w[i];
}
for(int i =1;i<=n;i++)
for(int j = 0;j<=m;j++) {
f[i][j] = f[i-1][j];
if(v[i] <= j) f[i][j] = max(f[i][j], f[i-1][j-v[i]] + w[i]);
}
cout<< f[n][m] << endl;
return 0;
}
dp-滚动数组
每一行的状态只与前一行的状态有关,所以可以用两个一维数组交替记录就可以实现。细想一下,每一行的状态只与前一行的当前位置向前有关,如果从后往前更新,就不会覆盖掉前一个状态,所以一个一维数组就可以实现。
#include <bits/stdc++.h>
using namespace std;
const int N = 1010;
int f[N], v[N], w[N];
int main(){
int n, m;
cin >> n >> m;
for(int i = 1; i<=n; i++) cin >> v[i] >> w[i];
for(int i = 1; i <= n; i ++)
for(int j = m; j >= v[i]; j --)
f[j] = max(f[j], f[j - v[i]] + w[i]);
cout<< f[m] << endl;
return 0;
}
总结
01背包题目常见,毕竟容易想到。y总说了
dp就得多练