代码随想录算法训练营Day7 | 454.四数相加II、383. 赎金信、15. 三数之和、18. 四数之和、总结 | Python | 个人记录向

454.四数相加II

代码随想录：454.四数相加II
Leetcode：454.四数相加II

做题

无简化思路。

看文章

遍历数组①②，将两数和存入字典；遍历数组③④，判断如果0 - 两数和在字典中，则count + 1。

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

时间复杂度: O(n^2)
空间复杂度: O(n^2)，最坏情况下A和B的值各不相同，相加产生的数字个数为 n^2

383. 赎金信

代码随想录：383. 赎金信
Leetcode：383. 赎金信

做题

from collections import defaultdict

class Solution:
    def canConstruct(self, ransomNote: str, magazine: str) -> bool:
        a = defaultdict(int) # 必须设置默认值，此时默认值为0
        for i in magazine:
            a[i] += 1
        
        for j in ransomNote:
            if a[j] > 0:
                a[j] -= 1
            else:
                return False
        return True

时间复杂度: O(n)
空间复杂度: O(1)

看文章

解法很多，具体见代码随想录。

15. 三数之和

代码随想录：15. 三数之和
Leetcode：15. 三数之和

做题

只能想到暴力求解，且不会去重。

看文章

使用双指针

先对数组进行由大到小排序。再定义3个指针：i、left、right，i遍历列表，left是i+1，right是len() - 1。遍历不重复的i，如果sum < 0，则left往右移动，如果sum > 0，则right往右移动。遇到重复的就跳过。

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
            
            # 跳过相同的元素以避免重复
            # 如果是判断i和i+1，就错了，因为i+1应该是left而不是i
            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

时间复杂度: O(n^2)
空间复杂度: O(1)

使用字典

去重思路比较复杂。

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        result = []
        nums.sort()
        # 找出a + b + c = 0
        # a = nums[i], b = nums[j], c = -(a + b)
        for i in range(len(nums)):
            # 排序之后如果第一个元素已经大于零，那么不可能凑成三元组
            if nums[i] > 0:
                break
            if i > 0 and nums[i] == nums[i - 1]: #三元组元素a去重
                continue
            d = {}
            for j in range(i + 1, len(nums)):
                if j > i + 2 and nums[j] == nums[j-1] == nums[j-2]: # 三元组元素b去重
                    continue
                c = 0 - (nums[i] + nums[j])
                if c in d:
                    result.append([nums[i], nums[j], c])
                    d.pop(c) # 三元组元素c去重
                else:
                    d[nums[j]] = j
        return result

时间复杂度: O(n^2)
空间复杂度: O(n)，额外的 set 开销

18. 四数之和

代码随想录：18. 四数之和
Leetcode：18. 四数之和

做题

时间紧，无思路。

看文章

使用双指针

与三数之和类似，区别在于：第2个数的去重。去重思路可以用第1个数的方法。

class Solution:
    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        nums.sort()
        n = len(nums)
        result = []
        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

时间复杂度: O(n^3)
空间复杂度: O(1)

使用字典

这种处理也挺有意思，而且看起来比较简洁。

class Solution(object):
    def fourSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        # 创建一个字典来存储输入列表中每个数字的频率
        freq = {}
        for num in nums:
            freq[num] = freq.get(num, 0) + 1
        
        # 创建一个集合来存储最终答案，并遍历4个数字的所有唯一组合
        ans = set()
        for i in range(len(nums)):
            for j in range(i + 1, len(nums)):
                for k in range(j + 1, len(nums)):
                    val = target - (nums[i] + nums[j] + nums[k])
                    if val in freq:
                        # 确保没有重复
                        count = (nums[i] == val) + (nums[j] == val) + (nums[k] == val)
                        if freq[val] > count:
                            ans.add(tuple(sorted([nums[i], nums[j], nums[k], val])))
        
        return [list(x) for x in ans]

哈希表总结

代码随想录：哈希表总结

以往忽略的知识点小结

• 用target - 当前和，来找最后一个数
• 用双指针解决n数之和的问题
• n数之和的去重：最后两个数可以用双指针，前面n - 2个数可以与前一个数比较（ i 和 i - 1 比较）
• 灵活使用字典

个人体会

完成时间：3h30min。
心得：题比较难，时间也比较紧张，思路和细节都需要学习。

​