题目描述:
给你一个整数数组 nums 和一个整数 k 。你需要找到 nums 中长度为 k 的 子序列 ,且这个子序列的 和最大 。
请你返回 任意 一个长度为 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
-105 <= nums[i] <= 105
1 <= k <= nums.length
方法1:
主要思路:解题链接汇总
(1)使用堆获得最大的k个数字;
(2)将这k个数字对应的索引进行排序;
(3)使用排序后的索引从数组中取出这k个数字;
class Solution {
public:
vector<int> maxSubsequence(vector<int>& nums, int k) {
priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>> p;
vector<int>res;
for(int i=0;i<k;i++){
p.push({nums[i],i});
}
for(int i=k;i<nums.size();i++){
if(nums[i]>p.top().first){
p.pop();
p.push({nums[i],i});
}
}
set<int> st;
while(p.size()>0){
st.insert(p.top().second);
p.pop();
}
for(auto i : st){
res.push_back(nums[i]);
}
return res;
}
};