内容仅供自己复习使用
这两个问题十分类似,一个是01背包。一个是完全背包。
整数划分因为一个数字可以多选,所以是完全背包问题。点菜问题只能选或者不选,所以是01背包问题。
未优化版本的小A点菜:
#include<bits/stdc++.h>
using namespace std;
const int N = 101;
int a[N];
int dp[N][10001];
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i = 1 ; i <= n ; i++) scanf("%d",&a[i]);
for(int i = 0 ; i <= n ; i++) dp[i][0] = 1;
for(int i = 1 ; i <= n ; i++)
{
for(int j = 1 ; j <= m ; j++)
{
dp[i][j] = dp[i-1][j];
if(j >= a[i]) dp[i][j] = dp[i][j] + dp[i-1][j-a[i]];
}
}
printf("%d",dp[n][m]);
return 0;
}
一维数组优化:
#include<bits/stdc++.h>
using namespace std;
const int maxn=10000+10;
int v[maxn],f[maxn];
int main(){
int n,m;
cin>>n>>m;
f[0]=1;
for(int i=1;i<=n;++i)
cin>>v[i];//读入 价值
for(int i=1;i<=n;++i)
for(int j=m;j>=v[i];--j)
f[j]+=f[j-v[i]];//现在的花费+=我不点这个菜的时候的花费
cout<<f[m]<<endl;最后把最后一个点的花费输出来就可以了
return 0;
}
整数划分:
#include<bits/stdc++.h>
using namespace std;
const int N = 1010 , mod = 1e9+7;
int dp[N][N];
int main()
{
int n;
cin>>n;
for(int i = 0 ; i <=n ; i++) dp[i][0] = 1;
for(int i = 1 ; i <= n ; i++)
{
for(int j = 0 ; j <= n ; j++)
{
dp[i][j] = dp[i-1][j] % mod;
if(j>=i) dp[i][j] = (dp[i-1][j] + dp[i][j-i]) % mod;
}
}
cout<<dp[n][n];
return 0;
}
一维数组优化:
#include<bits/stdc++.h>
using namespace std;
const int N = 1001;
int mod = 1e9+7;
int a[N];
int dp[N];
int main()
{
int n;
cin>>n;
dp[0] = 1;
for(int i = 1 ; i <= n ; i++)
{
for(int j = i ; j <= n; j++)
{
dp[j] = dp[j] + dp[j-i];
dp[j] %= mod;
}
}
cout<<dp[n];
return 0;
}