双指针
283.移动零
https://leetcode.cn/problems/move-zeroes/
数组 双指针 原地算法
给定一个数组
nums
,编写一个函数将所有0
移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。
我的思路:定义快慢指针,当数组中元素不为0时,快慢指针同时前进。当元素为0时,快指针前进一步,此时慢指针指向的值为0,快指针继续前进,直到指向的值不为0,快慢指针的值交换(将0移到后面),慢指针前进一步。直到快指针遍历完整个数组。
class Solution:
def moveZeroes(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
n = len(nums)
left, right = 0, 0 # 快慢指针
while right < n:
if nums[right] != 0:
# 其实相当于nums[left] == 0,交换0到后面
nums[left], nums[right] = nums[right], nums[left]
left += 1
right += 1
167. 两数之和 II - 输入有序数组
https://leetcode.cn/problems/two-sum-ii-input-array-is-sorted/
数组 双指针 二分查找
设计的解决方案必须只使用常量级的额外空间。
方法一:哈希表
达不到 O ( 1 ) O(1) O(1)的空间复杂度
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
res = {}
for idx, val in enumerate(numbers):
if target - val not in res:
res[val] = idx + 1
else:
return [res[target - val], idx + 1]
复杂度分析
- 时间复杂度: O ( N ) O(N) O(N),其中 N 是数组中的元素数量。对于每一个元素 x,我们可以 O ( 1 ) O(1) O(1) 地寻找 target - x。
- 空间复杂度: O ( N ) O(N) O(N),其中 N 是数组中的元素数量。主要为哈希表的开销。
方法二:二分查找
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
# 二分查找
n = len(numbers)
for i in range(n):
left, right = i+1, n - 1
while left <= right:
mid = (left + right) // 2
if numbers[mid] == target - numbers[i]:
return [i+1, mid+1]
elif numbers[mid] > target - numbers[i]:
right = mid - 1
else:
left = mid + 1
return [-1,-1]
复杂度分析
- 时间复杂度: O ( n log n ) O(n \log n) O(nlogn),其中 n 是数组的长度。需要遍历数组一次确定第一个数,时间复杂度是 O ( n ) O(n) O(n),寻找第二个数使用二分查找,时间复杂度是 O ( log n ) O(\log n) O(logn),因此总时间复杂度是 O ( n log n ) O(n \log n) O(nlogn)。
- 空间复杂度: O ( 1 ) O(1) O(1)
方法三:双指针(最优)
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
# 双指针
n = len(numbers)
left, right = 0, n - 1
while left <= right:
Sum = numbers[left] + numbers[right]
if Sum == target:
return [left+1, right+1]
elif Sum < target:
left += 1
else:
right -= 1
return [-1,-1]
复杂度分析
- 时间复杂度: O ( n ) O(n) O(n),其中 n 是数组的长度。两个指针移动的总次数最多为 nn 次。
- 空间复杂度: O ( 1 ) O(1) O(1)。