题目描述
求01背包前k优解的价值和
输入输出格式
输入格式:
第一行三个数K、V、N
接下来每行两个数,表示体积和价值
输出格式:
前k优解的价值和
输入输出样例
输入样例#1:
2 10 5
3 12
7 20
2 4
5 6
1 1
输出样例#1:
57
说明
对于100%的数据,Kle 50,Vle 5000,Nle 200K≤50,V≤5000,N≤200
思路:求01背包的前k优解的模板题。
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define MAXN 210 using namespace std; int K,V,N; int ans,cur; int f[5010][51]; int v[MAXN],val[MAXN]; int main(){ scanf("%d%d%d",&K,&V,&N); for(int i=1;i<=N;i++) scanf("%d%d",&v[i],&val[i]); for(int i=1;i<=N;i++){ for(int j=min(V,cur+v[i]);j>=v[i];j--){ int A[51],B[51]; for(int kk=1;kk<=K;kk++){ A[kk]=f[j-v[i]][kk]||j==v[i]&&kk==1?f[j-v[i]][kk]+val[i]:0; B[kk]=f[j][kk]; } int a=1,b=1,c=1; A[K+1]=-1,B[K+1]=-1; while(c<=K&&(A[a]!=-1||B[b]!=-1)){ if(A[a]>B[b]) f[j][c]=A[a++]; else f[j][c]=B[b++]; c++; } } cur=min(V,cur+v[i]); } for(int i=1;i<=K;i++) ans+=f[V][i]; cout<<ans; }