454.四数相加II,383. 赎金信,15. 三数之和 ,18. 四数之和
454. 四数相加 II
给你四个整数数组 n u m s 1 nums1 nums1、 n u m s 2 nums2 nums2、 n u m s 3 nums3 nums3 和 n u m s 4 nums4 nums4 ,数组长度都是 n n n ,请你计算有多少个元组 ( i , j , k , l ) (i, j, k, l) (i,j,k,l) 能满足:
- 0 < = i , j , k , l < n 0 <= i, j, k, l < n 0<=i,j,k,l<n
- n u m s 1 [ i ] + n u m s 2 [ j ] + n u m s 3 [ k ] + n u m s 4 [ l ] = = 0 nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0 nums1[i]+nums2[j]+nums3[k]+nums4[l]==0
示例 1:
输入:
n
u
m
s
1
=
[
1
,
2
]
,
n
u
m
s
2
=
[
−
2
,
−
1
]
,
n
u
m
s
3
=
[
−
1
,
2
]
,
n
u
m
s
4
=
[
0
,
2
]
nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
nums1=[1,2],nums2=[−2,−1],nums3=[−1,2],nums4=[0,2]
输出:2
解释:
两个元组如下:
- ( 0 , 0 , 0 , 1 ) − > n u m s 1 [ 0 ] + n u m s 2 [ 0 ] + n u m s 3 [ 0 ] + n u m s 4 [ 1 ] = 1 + ( − 2 ) + ( − 1 ) + 2 = 0 (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0 (0,0,0,1)−>nums1[0]+nums2[0]+nums3[0]+nums4[1]=1+(−2)+(−1)+2=0
- ( 1 , 1 , 0 , 0 ) − > n u m s 1 [ 1 ] + n u m s 2 [ 1 ] + n u m s 3 [ 0 ] + n u m s 4 [ 0 ] = 2 + ( − 1 ) + ( − 1 ) + 0 = 0 (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0 (1,1,0,0)−>nums1[1]+nums2[1]+nums3[0]+nums4[0]=2+(−1)+(−1)+0=0
示例 2:
输入:
n
u
m
s
1
=
[
0
]
,
n
u
m
s
2
=
[
0
]
,
n
u
m
s
3
=
[
0
]
,
n
u
m
s
4
=
[
0
]
nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0]
nums1=[0],nums2=[0],nums3=[0],nums4=[0]
输出:1
简单思路分为两组,HashMap 存一组,另一组和 HashMap 进行比对。这样的话情况就可以分为三种:
- HashMap 存一个数组,如 A。然后计算三个数组之和,如 BCD。时间复杂度为: O ( n ) + O ( n 3 ) O(n)+O(n^3) O(n)+O(n3),得到 O ( n 3 ) O(n^3) O(n3)。
- HashMap 存三个数组之和,如 ABC。然后计算一个数组,如 D。时间复杂度为: O ( n 3 ) + O ( n ) O(n^3)+O(n) O(n3)+O(n),得到 O ( n 3 ) O(n^3) O(n3)。
- HashMap存两个数组之和,如AB。然后计算两个数组之和,如 CD。时间复杂度为:
O
(
n
2
)
+
O
(
n
2
)
O(n^2)+O(n^2)
O(n2)+O(n2),得到
O
(
n
2
)
O(n^2)
O(n2)。
根据第二点我们可以得出要存两个数组算两个数组。以存 AB 两数组之和为例。首先求出 A 和 B 任意两数之和 sumAB,以 sumAB 为 key,sumAB 出现的次数为 value,存入 hashmap 中。然后计算 C 和 D 中任意两数之和的相反数 sumCD,在 hashmap 中查找是否存在 key 为 sumCD。算法时间复杂度为 O ( n 2 ) O(n2) O(n2)。
class Solution:
def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
res1 = dict()
res2 = dict()
n = len(nums1)
for i in range(n):
for j in range(n):
if nums1[i]+nums2[j] not in res1:
res1[nums1[i]+nums2[j]]=1
else:
res1[nums1[i]+nums2[j]]+=1
for i in range(n):
for j in range(n):
if nums3[i]+nums4[j] not in res2:
res2[nums3[i]+nums4[j]]=1
else:
res2[nums3[i]+nums4[j]]+=1
cnt = 0
for i in res1:
if 0 - i in res2:
cnt += res1[i] * res2[0-i]
return cnt
参考答案,采用一个字典,思路一样都是两两数组合并讨论。
class Solution(object):
def fourSumCount(self, nums1, nums2, nums3, nums4):
# 使用字典存储nums1和nums2中的元素及其和
hashmap = dict()
for n1 in nums1:
for n2 in nums2:
if n1 + n2 in hashmap:
hashmap[n1+n2] += 1
else:
hashmap[n1+n2] = 1
# 如果 -(n1+n2) 存在于nums3和nums4, 存入结果
count = 0
for n3 in nums3:
for n4 in nums4:
key = - n3 - n4
if key in hashmap:
count += hashmap[key]
return count
383. 赎金信
给你两个字符串:ransomNote 和 magazine ,判断 ransomNote 能不能由 magazine 里面的字符构成。如果可以,返回 true ;否则返回 false 。magazine 中的每个字符只能在 ransomNote 中使用一次。
示例 1
输入:ransomNote = “a”, magazine = “b”
输出:false
示例 2:
输入:ransomNote = “aa”, magazine = “ab”
输出:false
示例 3:
输入:ransomNote = “aa”, magazine = “aab”
输出:true
这道题简单对magazine 建立HashMap,最后判断 key, value分别为值和出现次数。对ransomNote 中每个元素遍历,HashMap出现则value-1,当value<0,返回false。遍历完成,value都大于等于0是返回True。
class Solution:
def canConstruct(self, ransomNote: str, magazine: str) -> bool:
res = dict()
res2 = dict()
for i in magazine:
if i not in res:
res[i] = 1
else:
res[i] += 1
for i in ransomNote:
if i not in res:
return False
res[i] -= 1
if res[i] < 0:
return False
return True
参考力扣官方题解,统计 magazine 中每个英文字母 a 的次数 cnt[a],再遍历统计 ransomNote 中每个英文字母的次数,如果发现 ransomNote 中存在某个英文字母 c 的统计次数大于 magazine 中该字母统计次数 cnt[c],则此时我们直接返回 false。
class Solution:
def canConstruct(self, ransomNote: str, magazine: str) -> bool:
if len(ransomNote) > len(magazine):
return False
return not collections.Counter(ransomNote) - collections.Counter(magazine)
15. 三数之和
给你一个整数数组 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 。
暴力解法,三层循环。
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
n = len(nums)
res = []
nums = sorted(nums)
for i in range(n):
for j in range(i+1,n):
for k in range(j+1,n):
if nums[i] + nums[j] + nums[k] == 0:
if [nums[i],nums[k],nums[j]] not in res:
res.append([nums[i],nums[k],nums[j]])
return res
参考代码,双指针,难点在于
- 排序后避免重复问题
- 跳过重复数字
- sum值窗口移动方式
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
result = []
nums.sort()
for i in range(len(nums)):
# 如果第一个元素已经大于0,不需要进一步检查
if nums[i] > 0:
return result
# 跳过相同的元素以避免重复
if i > 0 and nums[i] == nums[i - 1]:
continue
left = i + 1
right = len(nums) - 1
while right > left:
sum_ = nums[i] + nums[left] + nums[right]
if sum_ < 0:
left += 1
elif sum_ > 0:
right -= 1
else:
result.append([nums[i], nums[left], nums[right]])
# 跳过相同的元素以避免重复
while right > left and nums[right] == nums[right - 1]:
right -= 1
while right > left and nums[left] == nums[left + 1]:
left += 1
right -= 1
left += 1
return result
18. 四数之和
给你一个由 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]]
方法和上面一样,只是增加了一层,注意一些判断条件。
class Solution:
def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
result = []
if len(nums)<4:
return result
nums.sort()
for i in range(len(nums)):
if nums[i] > target and target>0:
break
if i>0 and nums[i]==nums[i-1]:
continue
for j in range(i+1, len(nums)):
if nums[i]+nums[j] > target and target>0:
continue
if j>i+1 and nums[j] == nums[j-1]:
continue
left = j+1
right = len(nums)-1
while left<right:
if nums[i]+nums[j]+nums[left]+nums[right]<target:
left += 1
elif nums[i]+nums[j]+nums[left]+nums[right]>target:
right -= 1
else:
result.append([nums[i],nums[j],nums[left],nums[right]])
while right > left and nums[right] == nums[right - 1]:
right -= 1
while right > left and nums[left] == nums[left + 1]:
left += 1
right -= 1
left += 1
return result