数据结构与算法之 leetcode 二分查找旋转数组搜索优化 跳过重复

数据结构与算法阿木 发布于 2025-07-12 10 次阅读


二分查找在旋转数组搜索中的应用与优化

在算法领域,二分查找是一种非常高效的搜索算法,其时间复杂度为O(log n),适用于有序数组。在实际应用中,我们可能会遇到旋转数组的情况,即数组中的元素被某个非递增的子序列分割,形成两个有序的子数组。在这种情况下,传统的二分查找方法可能无法直接应用。本文将探讨如何利用二分查找算法在旋转数组中进行搜索,并针对重复元素进行优化。

旋转数组概述

旋转数组是指将一个有序数组中的元素按照某个非递增的子序列进行旋转,形成一个新的数组。例如,有序数组[1, 2, 3, 4, 5, 6]经过旋转后可能变为[4, 5, 6, 1, 2, 3]。

传统二分查找在旋转数组中的应用

在旋转数组中,我们可以将问题分解为两个子问题:在左半部分查找和在右半部分查找。具体步骤如下:

1. 判断目标值是否在数组的左半部分。

2. 如果在左半部分,则在左半部分进行二分查找。

3. 如果不在左半部分,则在右半部分进行二分查找。

以下是一个简单的实现示例:

python

def search(nums, target):


left, right = 0, len(nums) - 1


while left <= right:


mid = (left + right) // 2


if nums[mid] == target:


return mid


判断目标值是否在左半部分


if nums[mid] >= nums[left]:


if nums[left] <= target < nums[mid]:


right = mid - 1


else:


left = mid + 1


else:


if nums[mid] < target <= nums[right]:


left = mid + 1


else:


right = mid - 1


return -1


针对重复元素的优化

在实际应用中,旋转数组中可能存在重复元素,这会导致传统二分查找方法失效。以下是一些优化策略:

1. 跳过重复元素:在二分查找过程中,如果发现当前元素与相邻元素相同,则跳过重复元素,继续进行二分查找。

2. 调整左右边界:在跳过重复元素后,需要根据跳过的元素数量调整左右边界。

以下是一个优化后的实现示例:

python

def search_optimized(nums, target):


left, right = 0, len(nums) - 1


while left <= right:


mid = (left + right) // 2


if nums[mid] == target:


return mid


跳过重复元素


while left < mid and nums[left] == nums[mid]:


left += 1


while right > mid and nums[right] == nums[mid]:


right -= 1


判断目标值是否在左半部分


if nums[mid] >= nums[left]:


if nums[left] <= target < nums[mid]:


right = mid - 1


else:


left = mid + 1


else:


if nums[mid] < target <= nums[right]:


left = mid + 1


else:


right = mid - 1


return -1


总结

本文介绍了二分查找在旋转数组搜索中的应用与优化。通过分析旋转数组的特性,我们提出了针对重复元素的优化策略,提高了算法的鲁棒性。在实际应用中,我们可以根据具体场景选择合适的二分查找方法,以实现高效的搜索。

扩展阅读

1. [LeetCode 33. 搜索旋转排序数组](https://leetcode-cn.com/problems/search-in-rotated-sorted-array/)

2. [LeetCode 81. 搜索旋转排序数组 II](https://leetcode-cn.com/problems/search-in-rotated-sorted-array-ii/)

3. [LeetCode 154. 寻找旋转排序数组中的最小值 II](https://leetcode-cn.com/problems/find-minimum-in-rotated-sorted-array-ii/)