0001.两数之和
题目:
给定一个整数数组 nums
和一个整数目标值 target
,请你在该数组中找出 和
为目标值 target
的那 两个 整数,并返回它们的数组下标。
思路1:
直接用两层循环暴力求解,算法复杂度为
O
(
n
2
)
O(n^2)
O(n2)。
第一层循环遍历数组中的所有元素,第二层循环找数组织该元素之后的元素是否等于
t
a
r
g
e
t
−
n
u
m
s
[
i
]
target-nums[i]
target−nums[i]即可。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
lens = len(nums)
find_tage = False
result = [0,0]
for i in range(lens-1):
cur = nums[i]
for j in range(i+1,lens):
if nums[j] == target - cur:
find_tage = True
result[1] = j
break
if find_tage:
result[0] = i
break
return result
思路2:
如果想要摆脱双层循环,实现低于
O
(
n
2
)
O(n^2)
O(n2)的算法复杂度,可以考虑使用字典的键值对进行求解。只使用一层循环,每次将
t
a
r
g
e
t
−
n
u
m
s
[
i
]
target-nums[i]
target−nums[i]作为键在字典中查找,找到则返回结果;未找到则将
n
u
m
s
[
i
]
nums[i]
nums[i]存入字典中。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
numDict = dict()
for i in range(len(nums)):
if (target - nums[i]) in numDict:
return numDict[target - nums[i]],i
numDict[nums[i]] = i
优化效果十分明显:
0027.移除元素
题目:
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
思路:
这道题可以采用双指针的知识去解决,让right指针指向当前未处理的元素,left指针的左侧则全为非val的元素,即left指针的左边全都满足要求,而left指针指向的元素直到right指针指向的元素,都为数值等于val的元素。
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
left = 0
right = 0
while right < len(nums):
if nums[right] != val:
nums[left],nums[right] = nums[right],nums[left]
left += 1
right += 1
return left
0026.删除有序数组中的重复项
题目:
给你一个 升序排列 的数组 nums ,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。
由于在某些语言中不能改变数组的长度,所以必须将结果放在数组nums的第一部分。更规范地说,如果在删除重复项之后有 k 个元素,那么 nums 的前 k 个元素应该保存最终结果。
将最终结果插入 nums 的前 k 个位置后返回 k 。
不要使用额外的空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
思路:
依旧使用双指针去解决,将left指针指向当前已删除重复元素的位置,即left指针指向的元素及之前的元素均未出现重复元素(即left指针及之前均满足条件),right指针指向当前已判断的元素。
当碰到left和right指向的元素重复时,right指针向后移动,而left不移动,此时left指向重复元素的第一个。
当left和right指向的元素不等时,需要将left的指针向后移动一位再将right指针指向的元素赋值过去(因为第一个重复元素还是要保留的)。
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
left = 0
right = 0
while right < len(nums):
if nums[left] == nums[right]:
right += 1
else:
left += 1
nums[left]= nums[right]
right += 1
return left+1
0015.三数之和
题目:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
思路:
最简单的思路就是三层循环暴力求解。注意到题目要求答案不包含重复的三元组,所以先将nums排序后再做判断即可。
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
result = []
nums.sort()
for i in range(len(nums)-2):
for j in range(i+1,len(nums)-1):
for k in range(j+1,len(nums)):
if nums[i]+nums[j]+nums[k] == 0:
temp = [nums[i],nums[j],nums[k]]
if temp not in result:
result.append(temp)
return result
但事实上,偷懒并没有好结果,超时了…
只能换种思路。
采用双指针的方法,先排序。对第一个元素a进行循环,即总共进行len(nums)次的循环。将left指针指向a的下一个元素,right指针指向最后一个元素,根据三个数之和与0的关系,决定两个指针的移动:
- 当三个数之和大于0时,将right指针往左移动;
- 当三个数之和小于0时,将right指针往右移动。
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
result = []
nums.sort()
n = len(nums)
for i in range(n):
if i > 0 and nums[i] == nums[i-1]:
continue
left = i + 1
right = n - 1
while left < right:
while left < right and left > i + 1 and nums[left] == nums[left-1]:
left += 1
while left <right and right < n - 1 and nums[right] == nums[right+1]:
right -= 1
if left < right and nums[i] + nums[left] + nums[right] == 0:
result.append([nums[i],nums[left],nums[right]])
left += 1
right -= 1
elif nums[i] + nums[left] + nums[right] < 0:
left += 1
else:
right -= 1
return result