P1164 小A点菜 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
题意:
思路:
01背包
设状态为dp[i][j],表示吃到第 i 种,剩下还能花 j 块钱
那么到达位置 i 时,需要对剩下的钱进行分类讨论:
如果剩下的钱还能买得起 a[i] ,那么就是买和不买的方案数加起来就行
如果剩下的钱买不起 a[i] ,那么就不买
Code :
#include <bits/stdc++.h>
using namespace std;
const int mxn=1e4+10;
int n,m;
int a[mxn],dp[110][mxn];
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
dp[0][0]=1;
for(int i=1;i<=n;i++){
for(int j=0;j<=m;j++){
if(j<a[i]) dp[i][j]=dp[i-1][j];
else if(j>=a[i])dp[i][j]=dp[i-1][j]+dp[i-1][j-a[i]];
}
}
printf("%d\n",dp[n][m]);
return 0;
}
注意到位置 i 的状态只由位置 i-1 的状态转移过来
因此可以用滚动数组:
#include <bits/stdc++.h>
using namespace std;
const int mxn=1e4+10;
int n,m;
int a[mxn],dp[mxn];
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
dp[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]];
}
}
printf("%d\n",dp[m]);
return 0;
}
总结:
线性dp可能要对其维度进行分类讨论,不一定所有的状态都由相同的转移方式转移