给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:
输入:nums = [3,2,4], target = 6
输出:[1,2]
示例 3:
输入:nums = [3,3], target = 6
输出:[0,1]
思路:
1、数据结构:列表 / 数组, hashset
2、算法:列表遍历
3、步骤:
a. 先创建一个hashset用于装nums里的数
b. for循环遍历nums(用到enumerate()函数),得到 索引 index 和 数 n
c. 每次遍历用 target - n 得到一个 temp, 若hashset里包含这个temp则直接返回两个索引
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
hashset = set()
for index, n in enumerate(nums):
temp = target - n
if temp in hashset:
return [index, nums.index(temp)]
hashset.add(n)
return None
给你四个整数数组 nums1
、nums2
、nums3
和 nums4
,数组长度都是 n
,请你计算有多少个元组 (i, j, k, l)
能满足:
0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
示例 1:
输入:nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2] 输出:2 解释: 两个元组如下: 1. (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0 2. (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0
示例 2:
输入:nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0] 输出:1
思路:
1、数据结构:collections.Counter()
2、算法:分而治之
3、步骤:
a. 将四个数组分成两对,AB和CD,令结果res = 0
b. 用collections.Counter将AB中的所有数相加后出现的次数存储
c. 遍历CD,若某个CD内出现的数的组合的相反数可以再AB中找到, res 就加上count中存储的该相反数的次数
class Solution:
def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
count12 = collections.Counter(u + v for u in nums1 for v in nums2)
res = 0
for x in nums3:
for y in nums4:
if -(x+y) in count12:
res += count12[-(x+y)]
return res
给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请你返回所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
解释:
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意,输出的顺序和三元组的顺序并不重要。
示例 2:
输入:nums = [0,1,1]
输出:[]
解释:唯一可能的三元组和不为 0 。
示例 3:
输入:nums = [0,0,0]
输出:[[0,0,0]]
解释:唯一可能的三元组和为 0 。
思路:
1、数据结构:列表 / 数组
2、算法:三指针遍历
3、步骤:
a. 一开始就要给数组升序排序。明确结果要返回二维列表,所以先创建 res = [ ] ,后续答案直接 res.append([..., ..., ... ])
b. 三颗指针,最左侧的 k 从左到右只遍历一次,右边的 i 和 j 则反复遍历 k 的右侧区间
c. 剪枝操作:若nums[k] > 0 则直接return res
d. 去重:若 nums[k-1] == nums[k] and k > 0,直接 continue 跳到下一个k
若nums[i] == nums[i-1] 则 i += 1(i 右移), 若nums[j] = nums[j+1], 则 j -= 1(j左移)
e. 由于一开始已经升序排序了数组,若s < 0 或 s > 0,则分别右移 i 或 左移 j 即可
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
res = []
nums.sort()
size = len(nums)
for k in range(0, size):
if nums[k] > 0:
return res
if nums[k-1] == nums[k] and k > 0:
continue
i, j = k + 1, size - 1
while i < j:
s = nums[k] + nums[i] + nums[j]
if s < 0:
i += 1
elif s > 0:
j -= 1
else:
res.append([nums[k], nums[i], nums[j]])
i += 1
j -= 1
while nums[i] == nums[i-1] and i < j:
i += 1
while nums[j] == nums[j+1] and i < j:
j -= 1
return res
给你一个由 n
个整数组成的数组 nums
,和一个目标值 target
。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]]
(若两个四元组元素一一对应,则认为两个四元组重复):
0 <= a, b, c, d < n
a
、b
、c
和d
互不相同nums[a] + nums[b] + nums[c] + nums[d] == target
你可以按 任意顺序 返回答案 。
示例 1:
输入:nums = [1,0,-1,0,-2,2], target = 0 输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
示例 2:
输入:nums = [2,2,2,2,2], target = 8 输出:[[2,2,2,2]]
思路:
上一题 15.三数之和 的进阶版, 关键是固定 左侧两个指针然后遍历右侧,关键在于剪枝和去重
class Solution:
def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
res = []
nums.sort()
size = len(nums)
for x in range(0, size - 3):
if nums[x] == nums[x-1] and x > 0:
continue
for y in range(x+1, size - 2):
if nums[y] == nums[y-1] and y > x + 1:
continue
i, j = y + 1, size - 1
while i < j:
s = nums[x] + nums[y] + nums[i] + nums[j]
if s < target:
i += 1
elif s > target:
j -= 1
else:
res.append([nums[x], nums[y], nums[i], nums[j]])
i += 1
j -= 1
while i < j and nums[i] == nums[i-1]:
i += 1
while i < j and nums[j] == nums[j+1]:
j -= 1
return res