https://www.acwing.com/problem/content/3/
朴素做法如下:
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=1010;
int v[N],w[N],f[N][N],n,m;
int main(void)
{
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++)
{
for(int k=0;k*v[i]<=j;k++)
{
f[i][j]=max(f[i][j],f[i-1][j-k*v[i]]+k*w[i]);
}
}
}
cout<<f[n][m]<<endl;
return 0;
}
朴素做法的时间复杂度太高了,会TLE。
f[i , j ] = max( f[i-1,j] , f[i-1,j-v]+w , f[i-1,j-2*v]+2*w , f[i-1,j-3*v]+3*w , .....)
f[i , j-v]= max( f[i-1,j-v] , f[i-1,j-2*v] + w , f[i-1,j-2*v]+2*w , .....)
由上两式,可得出如下递推关系:
f[i][j]=max(f[i-1][j] , f[i,j-v]+w)
优化后的代码如下:
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=1010;
int v[N],w[N],f[N][N],n,m;
int main(void)
{
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][j-v[i]]+w[i]);
}
}
cout<<f[n][m]<<endl;
return 0;
}
完全背包问题和01背包问题的递推公式对比
-
01背包:
f[i][j] = max(f[i][j],f[i-1][j-v[i]]+w[i]);
-
完全背包:
f[i][j] = max(f[i][j],f[i][j-v[i]]+w[i]);
通过观察你会发现:01背包的公式是由上一次的来更新的,而完全背包是由本次的来更新的。
一维优化代码如下: 体积从小到大枚举即可
状态f[j]定义:N 件物品,背包容量j下的最优解。
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=1010;
int v[N],w[N],f[N],n,m;
int main(void)
{
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=v[i];j<=m;j++)
{
f[j]=max(f[j],f[j-v[i]]+w[i]);
}
}
cout<<f[m]<<endl;
return 0;
}