跟着leedcode刷算法 -- 排序与检索2

193 阅读2分钟

这是我参与11月更文挑战的第26天,活动详情查看:2021最后一次更文挑战

image.png

题3

寻找峰值

峰值元素是指其值严格大于左右相邻值的元素。

给你一个整数数组 nums,找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回 任何一个峰值 所在位置即可。

你可以假设 nums[-1] = nums[n] = -∞ 。

你必须实现时间复杂度为 O(log n) 的算法来解决此问题。

示例 1:

  • 输入:nums = [1,2,3,1]

  • 输出:2

  • 解释:3 是峰值元素,你的函数应该返回其索引 2。 示例 2:

  • 输入:nums = [1,2,1,3,5,6,4]

  • 输出:1 或 5

  • 解释:你的函数可以返回索引 1,其峰值元素为 2;

  •   或者返回索引 5, 其峰值元素为 6。  

提示:

  • 1 <= nums.length <= 1000
  • -231 <= nums[i] <= 231 - 1
  • 对于所有有效的 i 都有 nums[i] != nums[i + 1] 相关标签
  • 数组
  • 二分查找

使用二分法进行查找

class Solution:
    def findPeakElement(self, nums: List[int]) -> int:
        if len(nums) == 1: 
            return 0

        left = 0
        right = len(nums) - 1

        while left < right:
            row = left + (right - left)//2
            if nums[row] > nums[row+1]:
                right = row
            else:
                left = row + 1
        return left

执行结果:

image.png

题4

寻找重复数

给定一个包含 n + 1 个整数的数组 nums ,其数字都在 1 到 n 之间(包括 1 和 n),可知至少存在一个重复的整数。

假设 nums 只有 一个重复的整数 ,找出 这个重复的数 。

你设计的解决方案必须不修改数组 nums 且只用常量级 O(1) 的额外空间。

示例 1:

  • 输入:nums = [1,3,4,2,2]

  • 输出:2 示例 2:

  • 输入:nums = [3,1,3,4,2]

  • 输出:3 示例 3:

  • 输入:nums = [1,1]

  • 输出:1 示例 4:

  • 输入:nums = [1,1,2]

  • 输出:1  

提示:

  • 1 <= n <= 105

  • nums.length == n + 1

  • 1 <= nums[i] <= n

  • nums 中 只有一个整数 出现 两次或多次 ,其余整数均只出现 一次   进阶:

  • 如何证明 nums 中至少存在一个重复的数字?

  • 你可以设计一个线性级时间复杂度 O(n) 的解决方案吗? 相关标签

  • 位运算

  • 数组

  • 双指针

  • 二分查找

思路: 集合

class Solution:
    def findDuplicate(self, nums: List[int]) -> int:
        res = set()
        for row in nums:
            if row in res:
                return row
            else:
                res.add(row)


执行结果:

image.png

题5

计算右侧小于当前元素的个数

给你一个整数数组 nums ,按要求返回一个新数组 counts 。数组 counts 有该性质: counts[i] 的值是  nums[i] 右侧小于 nums[i] 的元素的数量。

示例 1:

  • 输入:nums = [5,2,6,1]

  • 输出:[2,1,1,0] 解释:

  • 5 的右侧有 2 个更小的元素 (2 和 1)

  • 2 的右侧仅有 1 个更小的元素 (1)

  • 6 的右侧有 1 个更小的元素 (1)

  • 1 的右侧有 0 个更小的元素 示例 2:

  • 输入:nums = [-1]

  • 输出:[0] 示例 3:

  • 输入:nums = [-1,-1]

  • 输出:[0,0]  

提示:

  • 1 <= nums.length <= 105
  • -104 <= nums[i] <= 104 相关标签
  • 树状数组
  • 线段树
  • 数组
  • 二分查找
  • 分治
  • 有序集合
  • 归并排序

第一想法: 两个for循环判断是否比当前数据小,但是写完之后发现 超时啦。。。。

image.png

于是查找一些资料,发现一个很好用的模块: docs.python.org/3/library/b…

python 封装了很多的函数方法,可以直接用哦,这里就用到了 bisect.bisect_left 进行二分查找

于是就有了下面的代码:

class Solution:
    def countSmaller(self, nums: List[int]) -> List[int]:
        ans = []
        res = []
        for row in reversed(nums):
            idx = bisect.bisect_left(ans, row)
            res.append(idx)
            ans.insert(idx,row)
        return res[::-1]

执行结果:

image.png