代码随想录刷题第七天
四数相加II (LC 454)
给你四个整数数组 nums1
、nums2
、nums3
和 nums4
,数组长度都是 n
,请你计算有多少个元组 (i, j, k, l)
能满足:
0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
输入:nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
输出:2
解释:
两个元组如下:
(0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
(1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0
这道题相比相比于接下来的两道X数之和,更为简单,因为这道题四个数字分别来自四个不同的数组,题目说不需要考虑输出结果元组去重。并且,不需要返回找到的数字的下标,只需要返回可能得个数。因此解题步骤更为简单,与LC1 两数之和类似。步骤如下:
- 先定义一个字典
mapping
,key
为a+b
,value
为a+b
和的次数 - 两个
for loop
循环遍历a+b
的和的可能性和次数,放入mapping
中 - 两个
for loop
循环遍历c+d
;target = 0-(c+d)
, 寻找target是否在mapping
中,如果在count+=mapping[target]
代码实现:
class Solution(object):
def fourSumCount(self, nums1, nums2, nums3, nums4):
mapping = {}
count = 0
for num1 in nums1:
for num2 in nums2:
sum1 = num1+num2
if sum1 in mapping:
mapping[sum1]+=1
else:
mapping[sum1] = 1
for num3 in nums3:
for num4 in nums4:
target = 0-num3-num4
if target in mapping:
count+=mapping[target]
return count
赎金信 (LC 383)
给你两个字符串:ransomNote
和 magazine
,判断 ransomNote
能不能由 magazine
里面的字符构成。
如果可以,返回 true
;否则返回 false
。
magazine
中的每个字符只能在 ransomNote
中使用一次。
canConstruct(“a”, “b”) -> false
canConstruct(“aa”, “ab”) -> false
canConstruct(“aa”, “aab”) -> true
这道题与LC242 有效字母异位词类似,使用magazine
里面的字母能否组成ransom
. 因为所有的字符都是a-z里,只有26个字母,可以使用一个数组记录ransom字母出现的频率,步骤如下:
- 使用一个数组(26个字母大小)记录
ransom
字母出现的频率 - 循环
magazine
,把magazine
中字母从频率中减去 - 如果
mapping
剩下的频率全部小于零,则返回True
, 如果有一个大于零,则返回False
代码实现:
class Solution(object):
def canConstruct(self, ransomNote, magazine):
mapping = [0 for i in range(26)]
for char in ransomNote:
mapping[ord(char) - ord("a")]+=1
for char in magazine:
mapping[ord(char) - ord("a")]-=1
for i in range(len(mapping)):
if mapping[i] > 0:
return False
return True
三数之和 (LC 15) 很难
给你一个整数数组 nums
,判断是否存在三元组 [nums[i], nums[j], nums[k]]
满足 i != j、i != k 且 j != k
(不可重复) ,同时还满足 nums[i] + nums[j] + nums[k] == 0
。请
你返回所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
输入: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] 。
注意,输出的顺序和三元组的顺序并不重要。
这道题的难点在于:
- 需要返回找到的和的下标,而不是次数
- 需要去重,返回的元组可能有重复的下标,需要去重
使用双指针法:
- 先对数组进行从小到大的排序
- 一个
for
循环对i指针循环 - 在每次循环中,将left设置为i+1, right为len(nums)-1:
for (i...., i<nums.isze, i++):
# 大于零了, 因为nums排序了,将right--减小和
if nums[i]+nums[left]+nums[right]>0:
right--
# 小于零了, 因为nums排序了,将left--增加和
elif nums[i]+nums[left]+nums[right]<0:
left++
# 等于零
else:
res.append(nums[i], nums[left], nums[right])
- 去重!!! 如果
res
里已经有(-1,-2,3)
, 则不可以再有(-1,-2,3)
在for
循环中需要去重i, 如果当前i
和之前的i-1
值相同,则可以跳过当前循环,因为之前已经遍历过这个值了。当找到b,c
值后还要对b,c/left,right
去重。
代码实现:
class Solution(object):
def threeSum(self, nums):
res = []
nums.sort()
for i in range(len(nums)):
# 剪枝
if nums[i]>0:
break
# 去重i
if i>0 and nums[i] == nums[i-1]:
continue
# 错误去重a方法,将会漏掉-1,-1,2 这种情况
'''
if (nums[i] == nums[i + 1]) {
continue;
}
'''
left = i+1
right = len(nums)-1
while(right>left):
if nums[i]+nums[left]+nums[right]>0:
right-=1
elif nums[i]+nums[left]+nums[right]<0:
left+=1
else:
res.append([nums[i],nums[left],nums[right]])
# 去重bc
while(right>left and nums[left]==nums[left+1]):
left+=1
while(right>left and nums[right]==nums[right-1]):
right-=1
left+=1
right-=1
return res
四数之和 (LC 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
你可以按 任意顺序 返回答案 。
示例:
输入:nums = [1,0,-1,0,-2,2], target = 0
输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
这道题与第一题不同的是需要完成去重操作,与第三题的区别是和与target比较,而不是0.
这道题与三数之和非常相似,主要区别在去重和剪枝,并使用两个for循环+left+right 完成
剪枝区别:
nums[i]>target
不再是剪枝的条件,因为两个负数相加可以使sum
更小:
- 反例:
{-4,-1,0,0}; target = -5
,-4-1=-5
两个负数相加和更小
所以剪枝条件是:nums[i]>target and target>0 and nums[i]>0
代码实现:
class Solution(object):
def fourSum(self, nums, target):
res = []
nums.sort()
for i in range(len(nums)):
if nums[i] > target and nums[i] > 0 and target > 0:# 剪枝(可省)
break
if i>0 and nums[i] == nums[i-1]: #去重,从i>0开始,防止nums[-1]
continue
for j in range(i+1,len(nums)):
if nums[i] + nums[j] > target and target > 0: #剪枝(可省)
break
if j>i+1 and nums[j] == nums[j-1]: #去重,从j>i+1开始,防止nums[i](不属于j循环)
continue
left = j+1
right = len(nums)-1
while(right>left):
if (nums[i]+nums[j]+nums[left]+nums[right])>target:
right-=1
elif (nums[i]+nums[j]+nums[left]+nums[right])<target:
left+=1
else:
res.append([nums[i],nums[j],nums[left],nums[right]])
while (right>left and nums[left]==nums[left+1]):
left+=1
while (right>left and nums[right]==nums[right-1]):
right-=1
left+=1
right-=1
return res
总结:
今天的后两题难度很高,尤其是去重和剪枝的操作,需要二刷时在仔细研究整理。