题目描述
给你一个数组 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
Solution
这一题搞懂题目之其实很简单,我们先排序(升序就从后往前遍历,降序就从前往后遍历)总而言之,排序之后,从最大值向最小值的方向遍历,然后用temp
进行累加,直到满足temp * 2 > sum
,相当于temp > sum/2
,也就是temp
大于和的一半。说明,此时前面的数之和大于剩下的数之和。
public List<Integer> minSubsequence(int[] nums) {
// 计算数组的和 sum
int sum = Arrays.stream(nums).sum();
int n = nums.length;
// 升序排序
Arrays.sort(nums);
int temp = 0;
List<Integer> list = new ArrayList<>();
// 因为是升序排序,最大的在最后面,从后往前遍历
for (int i = n - 1; i >= 0; i--) {
list.add(nums[i]);
temp += nums[i];
if (temp * 2 > sum) // 满足 temp * 2 > sum 真的,说明 temp 超过总和的一半,比其他数之和更大
return list;
}
return list;
}