zl程序教程

您现在的位置是:首页 >  后端

当前栏目

【找到和最大的长度为 K 的子序列(2099-java)】

JAVA序列 最大 长度 找到
2023-09-27 14:29:28 时间

找到和最大的长度为 K 的子序列(2099-java)

给你一个整数数组 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] 。

public class LC241_2099_maxSubsequence {
    //排序
    public static int[] maxSubsequence(int[] nums, int k) {
        //1.临时数组
        int[] tmp = Arrays.copyOf(nums, nums.length);
        Arrays.sort(tmp);
        //2.统计个数
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = tmp.length - 1; i >= tmp.length - k; i--) {
            map.put(tmp[i], map.getOrDefault(tmp[i], 0) + 1);
        }
        //3.取数
        int[] ans = new int[k];
        int j = 0;
        for (int i = 0; i < nums.length; i++) {
            if (map.getOrDefault(nums[i], 0) > 0) {
                ans[j++] = nums[i];
                map.put(nums[i], map.get(nums[i]) - 1);
                if (j >= k) {
                    break;
                }
            }
        }
        return ans;
    }
    public static void main(String[] args) {
        int[] ints = maxSubsequence(new int[]{2, 1, 3, 3}, 2);
        for (int anInt : ints) {
            System.out.print(anInt + " ");
        }
    }
}