刷题使我快乐,满脸开心.jpg

  • 来源:力扣(LeetCode)
  • 链接:https://leetcode.cn/problems/create-maximum-number/
  • 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题目

给你一个整数数组 nums 和一个正整数 k ,返回长度为 k 且最具 竞争力nums 子序列。

数组的子序列是从数组中删除一些元素(可能不删除元素)得到的序列。

在子序列 a 和子序列 b 第一个不相同的位置上,如果 a 中的数字小于 b 中对应的数字,那么我们称子序列 a 比子序列 b (相同长度下)更具 竞争力 。 例如,[1,3,4][1,3,5] 更具竞争力,在第一个不相同的位置,也就是最后一个位置上, 4 小于 5

示例 1:

输入:nums = [3,5,2,6], k = 2
输出:[2,6]
解释:在所有可能的子序列集合 {[3,5], [3,2], [3,6], [5,2], [5,6], [2,6]} 中,[2,6] 最具竞争力。

示例 2:

输入:nums = [2,4,3,3,5,4,9,6], k = 4
输出:[2,3,3,4]

提示:

  • 1 <= nums.length <= 105
  • 0 <= nums[i] <= 109
  • 1 <= k <= nums.length

思路

虽然引入了一个新词,但是这道题跟402几乎是一样的,都是去掉给定数量的元素,找一个某种顺序(最大或者最小只是一个方向而已,代码思路都一样)的最终结果。

代码

func mostCompetitive(nums []int, k int) []int {
    removeNum := len(nums)-k
    res := make([]int, 0, k)
    for _, num := range nums {
        for len(res) > 0 && removeNum > 0 && num < res[len(res)-1] {
            res = res[:len(res)-1]
            removeNum--
        }
        res = append(res, num)
    }
    if len(res) > k {
        res = res[:k]
    }
    return res
}