LeetCode 2099. 找到和最大的长度为 K 的子序列

示例 1：
输入：nums = [2,1,3,3], k = 2
输出：[3,3]
解释：
子序列有最大和：3 + 3 = 6 。
示例 2：
输入：nums = [-1,-2,3,4], k = 3
输出：[-1,3,4]
解释：
子序列有最大和：-1 + 3 + 4 = 6 。
示例 3：
输入：nums = [3,4,3,3], k = 2
输出：[3,4]
解释：
子序列有最大和：3 + 4 = 7 。
另一个可行的子序列为 [4, 3] 。
提示：
1 <= nums.length <= 1000
-10^5 <= nums[i] <= 10^5
1 <= k <= nums.length

class Solution {
public:
    vector<int> maxSubsequence(vector<int>& nums, int k) {
        multiset<int> s;
        for(auto num : nums)
        {
            if(s.size() < k)
                s.insert(num);
            else if(*s.begin() < num)
            {
                s.erase(s.begin());
                s.insert(num);
            }
        } // 最大的 k 个数 在 s 中
        vector<int> ans;
        for(auto num : nums)
        {
            auto it = s.find(num);
            if(it != s.end())
            {
                ans.push_back(*it);
                s.erase(it);
            }
        }
        return ans;
    }
};