背包求第k优解 vijos 1412

#include<cstdio>
#include<iostream>
using namespace std;
const int M = 5000 + 20;
const int S = 220;
const int L = 60;
const int INF = -1e5;
int dp[M][L], n, k, m, v[S], w[S], q1[L], q2[L];
main()
{
scanf("%d %d %d", &k, &m, &n);
for(int i = 0; i < n; i++)
scanf("%d %d", &w[i], &v[i]);
for(int i = 0; i <= m; i++)
for(int j = 0; j <= k; j++)
dp[i][j] = INF;//先赋一个无限小的数，这样如果不是正好装满背包的权值也会变得无限小。
dp[0][0] = 0;
for(int i = 0; i < n; i++)
for(int j = m; j >= w[i]; j--)
{
int p;
for(p = 0; p < k; p++)//考虑01背包的dp[j] = max(dp[j], dp[j - w[i]] + v[i]),这里是每当添加一个新物品时，对所有的k进行一次这样的操作。
{
q1[p] = dp[j][p];
q2[p] = dp[j - w[i]][p] + v[i];
}
p = 0;
int l1 = 0, l2 = 0;
while(p < k)
{
if(q1[l1] > q2[l2])
dp[j][p] = q1[l1++];
else
dp[j][p] = q2[l2++];
p++;
}
}
int ans = 0;
for(int i = 0; i < k; i++)
ans += dp[m][i];
printf("%d\n", ans);
}