问题:求01背包前k优解的价值和;
方法:设方程为dp[i][j][k] , dp[i][j][k]代表前 i 个物品装在容量为 j 的背包中的第 k 优值。那么对于 dp [ i ] [ j ] 来说,由 dp[ i−1] [ j ] 和 dp[ i − 1 ] [ j − v ] + w 转移, 分别对应取与不取的两种情形 。而k优解,我们从 dp [ i − 1 ] [ j ] [ 1...k ] 和 dp[ i − 1] [ j − v ] [ 1...k ] + w 中选出前k大的给dp [ i ] [ j ] [ 1...k ] 。我们定义两个变量,逐一比较,t1代表不取的情况,t2代表取的情况,迭代,最后得到答案。跟归并排序的思想类似,在数组里面用两个变量分别标记,取走大的数,然后对应下标++。
tip:一定要设置初始的值,对于d[ 1 ] [ 0 ]来说是0,其他都是非法状态,设置为负无穷。
#include<bits/stdc++.h>
using namespace std;
const int maxn=5e3+100;
int d[55][maxn],tmp[55];
int main()
{
int k,V,n;
cin>>k>>V>>n;
memset(d,-0x3f,sizeof(d));
d[1][0]=0;
for(int i=1;i<=n;i++)
{
int v,w;
cin>>v>>w;
for(int j=V;j>=v;j--)
{
int t1=1,t2=1,t3=k,cnt=0;
while(t3--)
{
if(d[t1][j]>d[t2][j-v]+w)
tmp[++cnt]=d[t1++][j];
else tmp[++cnt]=d[t2++][j-v]+w;
}
for(int l=1;l<=k;l++)d[l][j]=tmp[l];
}
}
int ans=0;
for(int i=1;i<=k;i++)ans+=d[i][V];
printf("%d\n",ans);
return 0;
}
很妙的小dp :链接
#include<bits/stdc++.h>
using namespace std;
int d[1005];
int main()
{
int n,m;
cin>>n>>m;
d[0]=1;
for(int i=1;i<=n;i++)
{
int x;
cin>>x;
for(int j=m;j>=x;j--)
{
d[j]+=d[j-x];
}
}
cout<<d[m]<<'\n';
return 0;
}