python刷力扣454.四数相加

题目描述：

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

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

示例 1：

输入：nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
输出：2
解释：
两个元组如下：
1. (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
2. (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0

示例 2：

输入：nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0]
输出：1

解题思路：上来先搞一个四层循环

class Solution:
    def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
        count = 0
        N = len(nums1)
        for i in range(N):
            for j in range(N):
                for k in range(N):
                    for l in range(N):
                        if nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0:
                            count += 1
        return count

那必超时，换思路，加和问题试试哈希

先将前两个数组混合相加，将所有可能的结果及次数放入哈希表中

再将后两个数组混合相加同时判断是否和哈希表中的结果和为0，并记录和为0的次数输出。

class Solution:
    def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
        hashmap = dict()
        N = len(nums1)
        for i in range(N):
            for j in range(N):
                if nums1[i]+nums2[j] in hashmap:
                    hashmap[nums1[i]+nums2[j]] +=1 
                else:
                    hashmap[nums1[i]+nums2[j]] = 1
        
        count = 0
        for i in range(N):
            for j in range(N):
                if -nums3[i]- nums4[j] in hashmap:
                    count += hashmap[-nums3[i]- nums4[j]]
        return count