算法训练营 day7 ｜ 哈希表part02

454.四数相加II

class Solution:
    def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
        record = {}
        for i in nums1: 
            for j in nums2:
                if i + j in record:
                    record[i + j] += 1 # 记录两数之和出现的次数，key为和，value为出现次数
                else:
                    record[i + j] = 1
        
        count = 0
        for i in nums3:
            for j in nums4:
                key = 0 - (i + j)
                if key in record:
                    count += record[key]
        return count

总结：和昨天写的有效字母异位词那次有点类似。都是先存一下，然后再去查一下。这里是把前两个数组的加和以及出现次数存到dict中，后面查询后两个数组元素之和的相反数是否出现在之前的dict中。

383. 赎金信

题目链接

class Solution:
    def canConstruct(self, ransomNote: str, magazine: str) -> bool:
        record = {}
        for i in magazine:
            if i in record:
                record[i] += 1
            else:
                record[i] = 1
        for i in ransomNote:
            if i in record:
                record[i] -= 1
            else:
                record[i] = -1
        for i in ransomNote:
            if record[i] <= -1:
                return False
        return True

总结：这题整体思路和上题也挺像的。先存一下，然后再查一下。先存一下magaizin中每个字母(作为key)以及出现的次数(作为value)。然后处理ransonNote，如果出现一次就让dict中对应的value值-1，如果刚好满足题意那么dict中不会出现小于等于-1的value值。

这题也可以用hash法，开个数组去存，每次字母-‘a’对应ascii码值作为下标，数组里存该字母出现的次数。整体都差不多。偷懒就不实现了。

15. 三数之和

题目链接

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums.sort()
        res = []
        # right = len(nums) - 1 # 不能写在这
        
        for i in range(len(nums)):
            left = i + 1
            right = len(nums) - 1
            if nums[i] > 0:
                return res

            if i > 0 and nums[i] == nums[i - 1]:
                continue
            
            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[right] == nums[right - 1]:
                        right -= 1
                    while left < right and nums[left] == nums[left + 1]:
                        left += 1
                    right -= 1
                    left += 1
        return res

总结：这题看了提示使用双指针。然后自己写了半天，总是不能完全通过测试用例。后面找到原因是因以right的定义要写到for循环里面，这样才能保证下次for循环开始的时候right被重置为末尾。其他的就是注意考虑到边界条件过滤去重。

18. 四数之和

题目链接

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 = j + 1
                right = len(nums) - 1
                while right > left:
                    sum = nums[i] + nums[j] + nums[right] + nums[left]
                    if sum > target:
                        right -= 1
                    elif sum < target:
                        left += 1
                    else:
                        result.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
                        right -= 1
                        left += 1
        return result

总结：这题和三数之和的整体思路很像，都使用的是双指针。不过这里需要注意更多的细节。在剪枝和去重上稍微不同。