前言
dp(动态规划)在经过前两天的学习,掌握了如下几个简单模型,分享给大家。
简单分析
dp分为以下几步
第一步:建立状态数组;
第二步:确定初始状态;
第三步:建立状态转移方程;
第四步:找到题目所求状态;
模型
爬楼梯问题
P1192 台阶问题 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
step1:建立状态数组dp[i],表示到达第i层的种类数
step2:确定初始状态dp[0]=1;
step3: 建立状态转移方程dp[i]=(dp[i]+dp[j])%mod;j的范围是从max(0,i-k)到i;
step4: 最后输出dp[n];
解决代码
#include<iostream>
#include<algorithm>
using namespace std;
int mod=100003;
int main (){
int n,k;
cin>>n>>k;
int a[n+1];
a[0]=1;
for(int i=1;i<=n;i++){
a[i]=0;
for(int j=1;j<=k&&i-j>=0;j++){
a[i]=(a[i]+a[i-j])%mod;
}
}
cout<<a[n]<<endl;
return 0;
}
变式1:
假设不能连续走相同的步数,即不能连续走k步;
如果还用原来的数组去思考就会显得表述不够严谨,
因为dp[i]只能表示到i层的种类数,不能表示从哪一层上来的,
因此我们选择用二维数组来表示到达i层,还要走j步的种类数;
f[i][j]表示走j步到达i层的方案数;
状态转移方程为
f[i][j]=∑f[i-j][k](k≠j)=dp[i-j]-f[i-j][j]
解决代码
for(int i=1;i<=n;i++){
for(int j=1;j<=min(k,i);j++){
f[i][j]=dp[i-j]-f[i-j][j];
dp[i]+=f[i][j];
}
}
最长上升子序列
B3637 最长上升子序列 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
step1: 确定状态数组 dp[i];表示到第i个数字的最大子序列的长度;
step2: 确定初速状态 dp[0]=1;
step3: 建立状态转移方程
if(a[i]>a[j])
dp[i]=max(dp[i],dp[j]+1)
step4: 最后扫描以j结尾的的子序列长度中的最大值;
解决代码
#include<iostream>
#include<algorithm>
using namespace std;
int main (){
int n;
cin>>n;
int a[n+1],dp[n+1];
for(int i=1;i<=n;i++)
cin>>a[i];
dp[0]=1;
for(int i=1;i<=n;i++){
dp[i]=1;
for(int j=1;j<=i-1;j++){
if(a[i]>a[j])
dp[i]=max(dp[j]+1,dp[i]);
}
}
sort(dp+1,dp+1+n);
cout<<dp[n];
return 0;
}
01背包问题
P1049 [NOIP2001 普及组] 装箱问题 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
step1: 建立状态数组 dp[n+1][v+1],表示是否能够用前i个物品组成j空间的背包,1表示可以,0表示不可以;
step2: 确定初始状态,memset(dp,0,sizeof(dp));memset函数(数组地址,初始值,数量);
step3: 建立状态转移方程,dp[i][j]=max(dp[i-1][j],dp[i-1][j-a[i]]),(j>=a[i]);
step4: 最后找到dp[n][j]中的j;
解决代码
#include<iostream>
#include<cstring>
using namespace std;
int n, v;
int main() {
cin >> v >> n;
int dp[n+1][v+1];
memset(dp, 0, sizeof(dp)); // 使用memset初始化dp数组为0
for(int i=0;i<=n;i++){
dp[i][0]=1;
}
int a[n+1];
for(int i=1; i<=n; i++)
cin >> a[i];
for(int i=1; i<=n; i++) {
for(int j=1; j<=v; j++) {
if(dp[i-1][j])
dp[i][j] = 1;
if(j>=a[i]){
if(dp[i-1][j-a[i]])
dp[i][j]=1;
}
}
}
for(int j=v; j>=0; j--) {
if(dp[n][j]) {
cout << v - j;
return 0;
}
}
return 0;
}