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

最新推荐文章于 2024-10-27 12:59:29 发布

weixin_43270226

最新推荐文章于 2024-10-27 12:59:29 发布

阅读量647

点赞数 11

本文链接：https://blog.csdn.net/weixin_43270226/article/details/142317979

版权

数据结构专栏收录该内容

22 篇文章 0 订阅

订阅专栏

454.四数相加II，383. 赎金信，15. 三数之和，18. 四数之和

454. 四数相加 II

给你四个整数数组 $n u m s 1$ 、 $n u m s 2$ 、 $n u m s 3$ 和 $n u m s 4$ ，数组长度都是 $n$ ，请你计算有多少个元组 $(i, j, k, l)$ 能满足：

$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$

示例 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]$
输出：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$
$(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$

示例 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]$
输出：1

简单思路分为两组，HashMap 存一组，另一组和 HashMap 进行比对。这样的话情况就可以分为三种：

HashMap 存一个数组，如 A。然后计算三个数组之和，如 BCD。时间复杂度为： $O(n)+O(n^3)$ ，得到 $O(n^3)$ 。
HashMap 存三个数组之和，如 ABC。然后计算一个数组，如 D。时间复杂度为： $O(n^3)+O(n)$ ，得到 $O(n^3)$ 。
HashMap存两个数组之和，如AB。然后计算两个数组之和，如 CD。时间复杂度为： $O(n^2)+O(n^2)$ ，得到 $O(n^2)$ 。
根据第二点我们可以得出要存两个数组算两个数组。以存 AB 两数组之和为例。首先求出 A 和 B 任意两数之和 sumAB，以 sumAB 为 key，sumAB 出现的次数为 value，存入 hashmap 中。然后计算 C 和 D 中任意两数之和的相反数 sumCD，在 hashmap 中查找是否存在 key 为 sumCD。算法时间复杂度为 $O (n 2)$ 。

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