Day 7 哈希表part2

LC  454 四数相加

本题有点像两数之和，思路是先遍历A，B数组，将元素之和（a+b）放到一个字典中，键为两元素之和，值为出现的频次。遍历完之后再去遍历C,D数组，元素之和(c+d)若为-（a+b），那说明找到了对应的元组，并且元组个数为字典中的值。

对比两数之和再做几遍！

class Solution:
    def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) ->int:
        hash = {}
        for i in nums1:
            for j in nums2:
                hash[i+j] = hash.get(i+j,0) + 1
        cnt = 0
        for k in nums3:
            for l in nums4:
                if -(k+l) in hash:
                    cnt += hash.get(-(k+l))
        return cnt

LC 15 三数之和

本题的一大难点在于去重，例如对于[1,1,2,3,4],target=6来说，只有一个[1,2,4]是被接受的。

本题使用双指针，思路如下：

首先将数组排序（排序是为了后续去重），然后有一层for循环，i从下标0的地方开始，同时定一个下标left 定义在i+1的位置上，定义下标right 在数组结尾的位置上。

如果nums[i] + nums[left] + nums[right] > 0 就说明此时三数之和大了，因为数组是排序后了，所以right下标就应该向左移动，这样才能让三数之和小一些。

如果 nums[i] + nums[left] + nums[right] < 0 说明 此时 三数之和小了，left 就向右移动，才能让三数之和大一些，直到left与right相遇为止。

class Solution:
    def threeSum(self,nums: list[int]) -> list[list[int]]:
        res = []
        nums.sort()
        for i in range(len(nums)):
            left , right = i+1 , 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
'''
这里我犯了一个低级错误，elif写成if，结果一直报错，
一直以来都把elif和if混用，这里算提个醒，
三元判断，要是第二个写成了if，那else怎么算呢？就出错了！
'''
                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] == nums[right-1]:
                        right -= 1
                    '''
                    这里的判重逻辑是，若发现left下一个和它的值一样，直接跳过，同理right
                    '''
                    right -= 1
                    left += 1   
        return res   

LC 18 四数之和

本题跟三数之和思路差不多，先用两个for循环遍历确定a，b两数，双指针left，right确定另外两数，判断四数之和是否为target即可。

class Solution:
    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        res = []
        nums.sort()
        for a in range(len(nums)):
            if a > 0 and nums[a] == nums[a-1]:
                continue
            for b in range(a+1 , len(nums)):
                if b > a+1 and nums[b] == nums[b-1]:
                    continue
                left , right = b+1,len(nums)-1
                while left < right:
                    sum = nums[a] + nums[b] + nums[left] + nums[right]
                    if sum > target:
                        right -= 1
                    elif sum < target:
                        left += 1
                    else:
                        res.append([nums[a] , nums[b] , 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
        return res

区别再于以下两点，不能直接照抄三数之和的代码。

第一，此处无需判断nums[a]和target的大小，因为target可能为负，三数之和的target为0，对排序后的数组来说，若a>0,后续怎么加都不可能为0；但此处target若为负，即便第一个a大于了target，后续的b，left，right也可能为负，相加就等于target了

        for a in range(len(nums)):
            if nums[a] > target:
                return res #不必加这句话
            if a > 0 and nums[a] == nums[a-1]:
                continue

第二，b的去重逻辑：b是从a的下一位开始，但是不能写成if b>= a+1 and nums[b] == nums[b-1],因为若b取得到a+1，此时nums[b-1]其实就是a，而a，b他们分别可以为重复的数字，只是a、b遍历的过程中，不能取重复的数字而已。要理解好这句话，三号和四数之和的去重逻辑才能理解

for b in range(a+1 , len(nums)):
    if b > a+1 and nums[b] == nums[b-1]:
        continue