leetcode-非递增顺序的最小子序列

737 阅读1分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第5天,点击查看活动详情

题目描述

给你一个数组 nums,请你从中抽取一个子序列,满足该子序列的元素之和 严格 大于未包含在该子序列中的各元素之和。

如果存在多个解决方案,只需返回 长度最小 的子序列。如果仍然有多个解决方案,则返回 元素之和最大 的子序列。

与子数组不同的地方在于,「数组的子序列」不强调元素在原数组中的连续性,也就是说,它可以通过从数组中分离一些(也可能不分离)元素得到。

注意,题目数据保证满足所有约束条件的解决方案是 唯一 的。同时,返回的答案应当按 非递增顺序 排列。

 

示例 1:

输入:nums = [4,3,10,9,8]
输出:[10,9] 
解释:子序列 [10,9][10,8] 是最小的、满足元素之和大于其他各元素之和的子序列。但是 [10,9] 的元素之和最大。

示例 2:

输入:nums = [4,4,7,6,7]
输出:[7,7,6] 
解释:子序列 [7,7] 的和为 14 ,不严格大于剩下的其他元素之和(14 = 4 + 4 + 6)。因此,[7,6,7] 是满足题意的最小子序列。注意,元素按非递增顺序返回。

示例 3:

输入:nums = [6]
输出:[6]

  提示:

  • 1 <= nums.length <= 500
  • 1 <= nums[i] <= 100

思路

要让选取的数最小,那肯定要选择尽量大的数,刚好题目还要求返回按照非递增顺序,那么自然而然想到的方法就是先排序,然后逐渐取大的数,直到已经选择的数的和大于剩下数的和。在求已经选择数的和,因为每次循环都是多取1个数,所以可以采用累加的方式,不用每次都重新计算。同样的,对于剩下数的和,可以先求出整体的和,然后采取累减的方式。

Java版本代码

class Solution {
    public List<Integer> minSubsequence(int[] nums) {
        int total = Arrays.stream(nums).sum();
        Arrays.sort(nums);
        List<Integer> ans = new ArrayList<>();
        int sum = 0, left = total;
        for (int i = nums.length-1; i >= 0; i--) {
            sum += nums[i];
            left -= nums[i];
            ans.add(nums[i]);
            if (sum > left) {
                break;
            }
        }
        return ans;
    }
}