代码随想录刷题第七天|● 454.四数相加II ● 383. 赎金信 ● 15. 三数之和 ● 18. 四数之和 ● 总结

代码随想录刷题第七天

四数相加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 两数之和类似。步骤如下：

1. 先定义一个字典mapping, key为a+b, value为a+b和的次数
2. 两个for loop循环遍历a+b的和的可能性和次数，放入mapping中
3. 两个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字母出现的频率，步骤如下：

1. 使用一个数组（26个字母大小）记录ransom字母出现的频率
2. 循环magazine，把magazine中字母从频率中减去
3. 如果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] 。
注意，输出的顺序和三元组的顺序并不重要。

这道题的难点在于：

1. 需要返回找到的和的下标，而不是次数
2. 需要去重，返回的元组可能有重复的下标，需要去重

使用双指针法：

1. 先对数组进行从小到大的排序
2. 一个for循环对i指针循环
3. 在每次循环中，将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])

4. 去重！！！ 如果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

总结：

今天的后两题难度很高，尤其是去重和剪枝的操作，需要二刷时在仔细研究整理。