给你一个整数数组 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来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-subsequence-of-length-k-with-the-largest-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
实现思路:
建立一个结构体,存储数值和它所在的位置,进行两次排序,第一次对数值大小进行排序,第二次对原来所在的位置序号排序。
实现代码:
class Solution {
public:
struct pp{
int num;
int rank;
};
static bool cmp1(pp &a,pp &b){
return a.num>b.num;
}
static bool cmp2(pp a,pp b){
return a.rank<b.rank;
}
vector<int> maxSubsequence(vector<int>& nums, int k) {
vector<pp> a(nums.size());
for(int i=0;i<nums.size();i++){
a[i].rank=i;
a[i].num=nums[i];
}
sort(a.begin(),a.end(),cmp1);
sort(a.begin(),a.begin()+k,cmp2);
vector<int> ans;
for(int i=0;i<k;i++){
ans.push_back(a[i].num);
}
return ans;
}
};
这里不知道为什么cmp在类里返回值就一定要是static???