携手创作,共同成长!这是我参与「掘金日新计划 · 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;
};