二分查找
题目:704. 二分查找
给定一个 n
个元素有序的(升序)整型数组 nums
和一个目标值 target
,写一个函数搜索 nums
中的 target
,如果目标值存在返回下标,否则返回 -1
。
涉及数据结构
- 数组
我的思路
- 草稿伪代码+遍历查找
class Solution:
def search(self, nums: List[int], target: int) -> int:
target_list = []
for i in range(0,len(nums)):
if nums[i] == target:
return i
if not target_list :
return -1
解法优化
题目前提
- 有序数组,无重复元素
二分法思路
-
一分为二、比较middle、更新区间,不断将当前查找的范围缩小为原来的一半
-
时间复杂度:
O(logN)
-
易错点:左区间右边界和右区间左边界的更新
-
左闭右闭写法
class Solution:
def search(self, nums: List[int], target: int) -> int:
# 左闭右闭
left = 0
right = len(nums) - 1
while left <= right: #右闭
# middle = left + (right - left) // 2
middle = (left + right) // 2 # 向下取整
if nums[middle] > target:
right = middle -1 # 右闭,下次搜索不包括
elif nums[middle] < target:
left = middle + 1 # 左闭,下次搜索不包括
else:
return middle
return -1
- 左闭右开写法
class Solution:
def search(self, nums: List[int], target: int) -> int:
# 左闭右开
left = 0
right = len(nums)
while left < right: # 右开
middle = (left + right) // 2 # 向下取整
if nums[middle] > target:
right = middle # 右开,下次搜索包括
elif nums[middle] < target:
left = middle + 1 # 左闭,下次搜索不包括
else:
return middle
return -1
效果、心得:
- 运行时间降低
- 左闭右闭不容易出错
移除元素
题目:27. 移除元素
给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地
修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
涉及数据结构
- 数组
我的思路
- 暴力循环(python首层循环不能用for)
class Solution:
def removeElement(self, nums: list[int], val: int) -> int:
size = len(nums)
i = 0
while i in range(size): # 不能用for
# print('before if, i = ', i)
# print('nums[i] = ' , nums[i])
if nums[i] == val:
for j in range(i + 1, size):
nums[j-1] = nums[j]
i = i-1
# print('after if, i = ' , i)
size = size -1
# print('size = ' , size)
# print('nums = ',nums)
i = i+1
return size
解法
题目前提
- 数组不能删除只能覆盖
- 不能使用额外的数组空间
双指针思路
-
快慢指针
- 快指针:指向新数组需要的元素,新数组就是不含有目标元素的数组;
- 慢指针:指向更新 新数组下标的位置
-
只有不等于val,才是新数组需要的元素;把快指针获取到的值赋给慢指针所在位置
-
时间复杂度:
O(N)
-
快慢指针
class Solution:
def removeElement(self, nums: list[int], val: int) -> int:
slow = 0
for fast in range(0, len(nums)):
if nums[fast] != val:
nums[slow] = nums[fast]
slow += 1
return slow
效果、心得:
- 运行时间降低
- 暴力解法直观,但双指针一开始上手不好理解,但跟着逐步演示遍历过程,其实很好理解