给你一个整数数组 nums 和一个整数 k 。你需要找到 nums 中长度为 k 的 子序列 ,且这个子序列的 和最大 。
请你返回 任意 一个长度为 k 的整数子序列。
子序列 定义为从一个数组里删除一些元素后,不改变剩下元素的顺序得到的数组。
示例一:
输入:nums = [2,1,3,3], k = 2 输出:[3,3] 解释: 子序列有最大和:3 + 3 = 6 。
示例二:
输入:nums = [-1,-2,3,4], k = 3 输出:[-1,3,4] 解释: 子序列有最大和:-1 + 3 + 4 = 6 。
示例三:
输入: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
解题思路:
排序然后按k截取后半部分,用hashmap存储。key为值,value为个数。然后for循环遍历原数组,每取一个,value-1,最后就可以按顺序取出来了。代码如下:
class Solution {
public int[] maxSubsequence(int[] nums, int k) {
if(k == nums.length){
return nums;
}
int newNums[] = new int[nums.length];
for (int i = 0; i < nums.length; i++) {
newNums[i] = nums[i];
}
Arrays.sort(newNums);
HashMap<Integer, Integer> map = new HashMap<>();
for (int i = newNums.length - k; i < newNums.length; i++) {
map.put(newNums[i],map.getOrDefault(newNums[i],0) + 1);
}
int[] res = new int[k];
int index = 0;
for (int i = 0; i < nums.length; i++) {
if(map.getOrDefault(nums[i],0) > 0 && index < k){
map.put(nums[i],map.get(nums[i])-1);
res[index++] = nums[i];
}
}
return res;
}
}
提交截图:
总结:我太菜了