1403. 非递增顺序的最小子序列
给你一个数组 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] 是满足题意的最小子序列。注意,元素按非递增顺序返回。
思路
- 在本题中要求抽取一个子序列,满足该子序列的元素之和严格大于未包含在该子序列中的各元素之和,则说明当前子序列的元素之和大于整个数组元素和的一半。
- 此外,要求序列长度最小,元素之和最大,考虑使用贪心法。考虑取出的元素尽可能大,才能保证长度小,元素之和最大。
- 综上所述,将数组从大到小排序,直到取出元素之和大于元素的一半。
解法
func minSubsequence(nums []int) []int {
sum:=0
for i:=0;i<len(nums);i++{
sum=sum+nums[i]
}
targtet:=sum/2
sort.Slice(nums,func(i, j int)bool{
return nums[i]>nums[j]
})
cur:=0
for i:=0;i<len(nums);i++{
cur+=nums[i]
if cur>targtet{
return nums[:i+1]
}
}
return []int{-1,-1}
}
结果
思路
-同样使用贪心法,相比上中解法,不需要求平均值,使用当前的元素和剩余的元素和比较即可,只要当前的元素和大于剩余的元素和,这样空间优化更好。
解法
func minSubsequence(nums []int) []int {
sum:=0
for i:=0;i<len(nums);i++{
sum=sum+nums[i]
}
sort.Slice(nums,func(i, j int)bool{
return nums[i]>nums[j]
})
cur:=0
for i:=0;i<len(nums);i++{
cur+=nums[i]
if cur>sum-cur{
return nums[:i+1]
}
}
return []int{-1,-1}
}
结果