704. 二分查找
难度:☆2
数组。前提条件是(1)有序数组,(2)没有重复项。如果碰到在有序数组中查找,可以想到用二分法;没有重复项才能输出唯一的索引。
需要注意坚持循环不变量,若是左闭右闭区间:right = len(nums) - 1;while left <= right: 要使用 <=;if nums[middle] > target: 时 right 要更新为 middle - 1。若是左闭右开区间:right = len(nums);while left < right: 要使用 <;right = middle。
a. 左闭右闭区间 [left, right]
class Solution:
def search(self, nums: List[int], target: int) -> int:
left = 0
right = len(nums) - 1
while left <= right:
index = (left + right) // 2
if nums[index] == target:
return index
elif nums[index] < target:
left = index + 1
elif nums[index] > target:
right = index - 1
return -1
第一次写,本能的想法是左闭右闭区间,左闭右开区间是在提示下写出来的。
b. 左闭右开区间 [left, right)
class Solution:
def search(self, nums: List[int], target: int) -> int:
left = 0
right = len(nums)
while left < right:
middle = left + (right - left) // 2
if nums[middle] > target:
right = middle
elif nums[middle] < target:
left = middle + 1
else:
return middle
return -1
另外,middle = left + (right - left) // 2 可以防止整数溢出。
Python Doc: 运算符 /
(除) 和 //
(整除) 将输出其参数的商。两个数字参数将先被转换为相同类型。整数相除会输出一个 float 值,整数相整除的结果仍是整数;整除的结果就是使用 'floor' 地板函数进行算术除法的结果。除以零的运算将引发 [ZeroDivisionError](<https://docs.python.org/zh-cn/3/library/exceptions.html#ZeroDivisionError>)
异常。
27. 移除元素
难度:☆3
数组。相向双指针 / 快慢双指针,首先想到相向双指针。注意循环在什么情况下结束,left <= right还是left < right。
a. 相向双指针1
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
left = 0
right = len(nums) - 1
while left <= right:
if nums[left] == val:
nums[left] = nums[right]
right -= 1
else:
left += 1
return left
b. 相向双指针2
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
left = 0
right = len(nums) - 1
while left <= right:
while left <= right and nums[right] == val:
right -= 1
while left <= right and nums[left] != val:
left += 1
if left < right:
nums[left] = nums[right]
left += 1
right -= 1
return left
c. 快慢双指针
快慢双指针挺不好想的,需要一点计算思维。快指针指向替换元素的来源,一定不等于val;慢指针指向被替换的元素,可能是val,也可能不是val(这个没想到)。最后返回慢指针的索引,就是移除元素后新数组的长度。
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
slow = 0
for fast in range(len(nums)):
if nums[fast] != val:
nums[slow] = nums[fast]
slow += 1
return slow