【力扣算法题每日一刷】1403. 非递增顺序的最小子序列(简单)

93 阅读1分钟

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

一、题目描述

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

来源:力扣(LeetCode)

二、思路分析

题目要求返回一个子序列,该子序列的元素之和 严格 大于未包含在该子序列中的各元素之和。

需要注意的是,子序列并不一定是连续子序列,也就是说子序列中的元素的顺序可以是随意的;同时,返回的答案应当按 非递增顺序 排列。

还需要注意另外一个条件:如果存在多个解决方案,只需返回 长度最小 的子序列。如果仍然有多个解决方案,则返回 元素之和最大 的子序列。

因此,我们需要尽可能少次数地找到最大的元素,总和(设为x)大于数组中不包含已找到的元素之和(设为y)。思路如下:

  • 将数组从小到大排序;
  • 计算nums数组的总和;
  • 定义ans数组存储子序列,sum存储遍历过程中计算的元素之和;
  • 倒着遍历排序过后的nums数组,每次遍历时sum加上当前元素,并且将当前元素存进ans序列数组中(因为nums数组已经排过序了,遍历到的元素一定是最大或较大的),只要 y < x就可以结束循环,return子序列数组了。

三、AC 代码

/**
* @param {number[]} nums
* @return {number[]}
*/
var minSubsequence = function(nums) {
    const total = _.sum(nums);
    nums.sort((a, b) => a - b);
    const ans = [];
    let sum = 0;
    for (let i = nums.length - 1; i >= 0; --i) {
        sum += nums[i];
        ans.push(nums[i]);
        if (total - sum < sum) {
            break;
        }
    }
    return ans;
};