AcWing 11. 直接套01背包模板2种写法
原题链接
中等
直接套用经典的01背包模板
#include<bits/stdc++.h>
using namespace std;
//这里没有说背包要装满
//也可以不装满
//++++++套用经典的01背包++++++++
int v[1005],w[1005];
int dp[1005][1005]={0},f[1005][1005]={0};
//f[N][N]={1}注意这种初始化方法是错误的
int main(){
int N,V;
cin>>N>>V;
for(int i=1;i<=N;i++){
cin>>v[i]>>w[i];
}
for(int i=0;i<=N;i++){
for(int j=0;j<=V;j++)
{
f[i][j]=1;
}
}
for(int i=1;i<=N;i++){
for(int j=1;j<=V;j++){
dp[i][j]=dp[i-1][j];
f[i][j]=f[i-1][j];
if(j>=v[i]){
int value = dp[i-1][j-v[i]]+w[i];
if(dp[i-1][j]<value){
dp[i][j]=value;
f[i][j] = f[i-1][j-v[i]];
}
else if(dp[i-1][j]==value){
//dp[i][j]=dp[i-1][j];
f[i][j]=(f[i-1][j]+f[i-1][j-v[i]])%1000000007;
}
}
}
}
cout<<f[N][V];
return 0;
}
滚动数组优化
#include<bits/stdc++.h>
using namespace std;
int v[1005],w[1005];
int dp[1005]={0},f[1005]={0};
int main(){
int N,V;
cin>>N>>V;
for(int i=1;i<=N;i++){
cin>>v[i]>>w[i];
}
for(int i=0;i<=V;i++){
f[i]=1;
}
for(int i=1;i<=N;i++){
for(int j=V;j>=v[i];j--){
int value = dp[j-v[i]]+w[i];
if(value>dp[j]){
dp[j]=value;
f[j]=f[j-v[i]];
}
else if(value==dp[j]){
dp[j]=value;
f[j]=(f[j]+f[j-v[i]])%1000000007;
}
}
}
cout<<f[V];
return 0;
}