代码随想录训练营 | 454.四数相加II 383. 赎金信 15. 三数之和 18. 四数之和

454.四数相加II

题目链接：

454. 四数相加 II - 力扣（LeetCode）

题目描述：

给你四个整数数组 nums1、nums2、nums3 和 nums4 ，数组长度都是 n ，请你计算有多少个元组 (i, j, k, l) 能满足：

• 0 <= i, j, k, l < n
• nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

解题思路：

哈希法：

使用count记录四数相加为零的次数，使用一个字典(num)存放nums1+nums2以及它的出现次数 ，若num3+nums4出现在字典(num)中出现，用count将字典中key对应的value统计出来。

代码：

class Solution:
    def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
        num = dict()
        count = 0
        for n1 in nums1:
            for n2 in nums2:
                if n1 + n2 in num:
                    num[n1 + n2] += 1
                else:
                    num[n1 + n2] = 1
        for n3 in nums3:
            for n4 in nums4:
                if -(n3 + n4) in num:
                    count += num[-(n3 + n4)]
        return count

383. 赎金信

题目链接：

383. 赎金信 - 力扣（LeetCode）

题目描述：

给你两个字符串：ransomNote 和 magazine ，判断 ransomNote 能不能由 magazine 里面的字符构成。

如果可以，返回 true ；否则返回 false 。

magazine 中的每个字符只能在 ransomNote 中使用一次。

解题思路：

分别用两个列表将ransomNote 和 magazine 中出现的字母次数统计下来，再判断magazine对应的列表中字母出现的次数是否都大于ranaomNote对应的列表，是返回True，否返回False

代码：

class Solution:
    def canConstruct(self, ransomNote: str, magazine: str) -> bool:
        nums1 = [0] * 26
        nums2 = [0] * 26
        for i in magazine:
            nums1[ord(i) - ord('a')] += nums1[ord(i) - ord('a')] + 1
        for i in ransomNote:
            nums2[ord(i) - ord('a')] += nums2[ord(i) - ord('a')] + 1
        for i in range(26):
            if nums1[i] < nums2[i]:
                return False
            else:
                nums1[i] -= 1
        else:
            return True

15. 三数之和

题目链接：

15. 三数之和 - 力扣（LeetCode）

题目描述：

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

注意：答案中不可以包含重复的三元组。

解题思路：

使用双指针法，最外面一层for循环，用来充当i的下标，里面一个left（充当j的下标）定义在i+1的位置上，一个right（充当k的下标）定义在数组最后，如果 nums[i] + nums[left] + nums[right] < 0 ，left 就向右移动，才能让三数之和大一些，。如果nums[i] + nums[left] + nums[right] > 0 ，right下标就应该向左移动，这样才能让三数之和小一些。直到left与right相遇为止。

需注意的是，答案中不可以包含重复的三元组，所以我们要对i，left，right对应的元素去重，即跳过相同的元素以免重复，并且不能有重复的三元组，但三元组内的元素是可以重复的，所以判断 nums[i] 与 nums[i-1]  是否相同，即判断比较它的前一个是否相同。

代码：

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        res = []
        nums.sort()
        for i in range(len(nums)):
            if i < 0:
                return res
            # 跳过相同的元素以避免重复
            if i > 0 and nums[i-1] == nums[i]:
                continue
            left = i + 1
            right = len(nums) - 1
            while left < right:
                sum_ = nums[i] + nums[left] + nums[right]
                if sum_ > 0:
                    right -= 1
                elif sum_ < 0:
                    left += 1
                else:
                    res.append( [nums[i], nums[left], nums[right]] )
                    # 跳过相同的元素以避免重复
                    while left < right and nums[left] == nums[left + 1]:
                        left += 1
                    # 跳过相同的元素以避免重复
                    while left < right and nums[right-1] == nums[right]:
                        right -= 1
                    left += 1
                    right -= 1
        return res

18. 四数之和

题目链接：

18. 四数之和 - 力扣（LeetCode）

题目描述：

给你一个由 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

你可以按 任意顺序 返回答案 。

解题思路：

该题和上一题15. 三数之和类似，都是使用双指针法，基本解法是在上一题的基础上再加一层for循环，两层for循环nums[i] + nums[j]为确定值，依然是循环内有left和right下标作为双指针，找出nums[k] + nums[i] + nums[left] + nums[right] == target的情况。

但是有一些细节需要注意，例如： 不要判断nums[k] > target 就返回了，三数之和 可以通过 nums[i] > 0 就返回了，因为 0 已经是确定的数了，四数之和这道题目 target是任意值。比如：数组是[-4, -3, -2, -1]，target是-10，不能因为-4 > -10而跳过。但是我们依旧可以去做剪枝，逻辑变成nums[i] > target && (nums[i] >=0 || target >= 0)就可以了。

代码：

class Solution:
    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        result = []
        nums.sort()
        n = len(nums)
        for i in range(n):
            if nums[i] > target and nums[i] > 0 and target > 0:# 剪枝（可省）
                break
            if i > 0 and nums[i] == nums[i-1]:# 去重
                continue
            for j in range(i+1, n):
                if nums[i] + nums[j] > target and target > 0: #剪枝（可省）
                    break
                if j > i+1 and nums[j] == nums[j-1]: # 去重
                    continue
                left, right = j+1, n-1
                while left < right:
                    s = nums[i] + nums[j] + nums[left] + nums[right]
                    if s == target:
                        result.append([nums[i], nums[j], nums[left], nums[right]])
                        while left < right and nums[left] == nums[left+1]:
                            left += 1
                        while left < right and nums[right] == nums[right-1]:
                            right -= 1
                        left += 1
                        right -= 1
                    elif s < target:
                        left += 1
                    else:
                        right -= 1
        return result