01背包第k优解-HDU2639

2023大厂真题提交网址(含题解):

www.CodeFun2000.com（http://101.43.147.120/）

最近我们一直在将收集到的机试真题制作数据并搬运到自己的OJ上，供大家免费练习,体会真题难度。现在OJ已录入50+道2023年最新大厂真题，同时在不断的更新。同时，可以关注"塔子哥学算法"公众号获得每道题的题解。

题意:

01背包第k（不同的）优解。

思路:

升维表示：$dp[i][j][k]$代表前$i$个物品,体积为$j$的第$k$优解。

转移变成合并两个有序数列。双指针或者排序即可。

AC代码:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define pb push_back
#define mp make_pair
#define vi vector<int>
#define vl vector<ll>
#define fi first
#define se second
const int maxn = 100 + 5;
const int maxv = 1000 + 5;
const int mod = 1e9 + 7;
pii a[maxn];
int dp[maxv][33];
int main()
{
    ios::sync_with_stdio(false);
    int t; cin >> t;
    while (t--){
        int n , v , num; cin >> n >> v >> num;
        for (int i = 1 ; i <= n ; i++){
            cin >> a[i].second;
        }
        for (int i = 1 ; i <= n ; i++){
            cin >> a[i].first;
        }
        memset(dp , 0 , sizeof dp);
        dp[0][1] = 0;
        for (int i = 1 ; i <= n ; i++){
            for (int j = v ; j >= a[i].first ; j--){
                vi tmp;
                for (int k = 1 ; k <= num ; k++){
                    if (~dp[j][k])
                        tmp.push_back(dp[j][k]);
                    if (~dp[j - a[i].first][k])
                        tmp.push_back(dp[j - a[i].first][k] + a[i].second);
                }
                sort(tmp.begin() , tmp.end());
                tmp.erase(unique(tmp.begin() , tmp.end()),tmp.end());
                reverse(tmp.begin(), tmp.end());
              //  for (auto g : tmp) cout << g << " ";
               // cout << endl;
                for (int k = 0 ; k < min((int)tmp.size() , num) ; k++) dp[j][k + 1] = tmp[k];
                //for (int k = tmp.size(); k < num ; k++) dp[j][k + 1] = -1;
            }
        }
       /* for (int i = 1 ; i <= num ; i++){
            cout << dp[v][i] << endl;
        }*/
        if (dp[v][num] == 0) cout << 0 << endl;
        else cout << dp[v][num] << endl;
    }
    return 0;
}