我今天智商可能不太够用。。。
背包K优解,我们考虑01背包最优解如何转移。在两种状态中取最优的。
对于k优解,我们可以多开一维记录。
转移的时候,我们开两个指针变量。记录现在哪个优,然后像归并排序一样往一个tmp数组里丢(不能现在改变值)。
然后就没什么了吧。。但我们因为要装满所以数组初始化为极小值,不能重复,所以只有f 0 1是0.
时间复杂度多了一层枚举k优。O(nmk)
#include<bits/stdc++.h>
using namespace std;
const int INF=1e9+7;
int f[5005][60],a[60],b[60],k,v,n;
struct data{
int pos,val;
}p[300];
int main(){
scanf("%d%d%d",&k,&v,&n);
for(int i=v;i>=0;i--){
for(int j=k;j>=1;j--)
f[i][j]=-INF;
}
f[0][1]=0;
for(int i=1;i<=n;i++){
scanf("%d%d",&p[i].pos,&p[i].val);
}
for(int i=1;i<=n;i++)
for(int j=v;j>=p[i].pos;j--){
int t1=1,t2=1;
for(int kk=1;kk<=k;kk++){
if(f[j][t1]>f[j-p[i].pos][t2]+p[i].val){
a[kk]=f[j][t1];
t1++;
}
else {
a[kk]=f[j-p[i].pos][t2]+p[i].val;
t2++;
}
}
for(int d=1;d<=k;d++)f[j][d]=a[d];
}
int ans=0;
for(int i=1;i<=k;i++){
ans+=f[v][i];
}
printf("%d\n",ans);
return 0;
}