DAY 1 代码随想录刷题
704.二分查找
题目
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
思路
题目直接点明要用二分法,但是要对数组的区间开闭的情况讨论。
1、左闭右闭
假设数组的区间为[left,right],那么取区间的中值middle,判断nums[middle]和targrt的大小,然后更新左/右边界。
class Solution:
def search(self, nums: List[int], target: int) -> int:
left, right = 0, len(nums) - 1
while left <= right:
middle = left + (right - left) // 2 #不使用(left+right)/2是防止数据可能溢出
if nums[middle] > target:
right = middle - 1 # target在左区间,所以[left, middle - 1]
elif nums[middle] < target:
left = middle + 1 # target在右区间,所以[middle + 1, right]
else:
return middle # 数组中找到目标值,直接返回下标
return -1 # 未找到目标值
2、左闭右开
由于右侧是开区间,因此left不能和right相等。
class Solution:
def search(self, nums: List[int], target: int) -> int:
left, right = 0, len(nums) # 右侧开,所以right可以取len(nums)
while left < right: # 因为left == right的时候,在[left, right)是无效的空间,所以使用 <
middle = left + (right - left) // 2
if nums[middle] > target:
right = middle # target 在左区间,更新右区间
elif nums[middle] < target:
left = middle + 1 # target 在右区间,更新左区间
else:
return middle
return -1
需要注意的是,根据区间的开闭情况不同,左右区间更新值是不一样的。
27. 移除元素
题目
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
思路:
1、双指针法
考虑使用双指针法(快慢指针法): 通过一个快指针和慢指针在一个for循环下完成两个for循环的工作。
定义快慢指针
快指针:寻找新数组的元素 ,新数组就是不含有目标元素的数组
慢指针:指向更新 新数组下标的位置
class Solution(object):
def removeElement(self, nums, val):
"""
:type nums: List[int]
:type val: int
:rtype: int
"""
#1、双指针思路 原地修改数组且额外空间O(1)
slow = 0 #慢指针
fast = 0 #快指针
length = len(nums)
while fast<length:
if nums[fast] != val:
nums[slow] = nums[fast]
# slow 用来收集不等于 val 的值,如果 fast 对应值不等于 val,则把它与 slow 替换
slow+=1
fast += 1
return slow
2、暴力解法
思路
使用两次for循环即可
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
i, l = 0, len(nums)
while i < l:
if nums[i] == val: # 找到等于目标值的节点
for j in range(i+1, l): # 移除该元素,后面元素向平移
nums[j - 1] = nums[j]
l -= 1 #数组总长度减1
i -= 1 #回到上一个位置
i += 1
return l