【代码随想录】刷题记录(17)-四数相加 II

题目描述：

给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ，使得 A[i] + B[j] + C[k] + D[l] = 0。

为了使问题简单化，所有的 A, B, C, D 具有相同的长度 N，且 0 ≤ N ≤ 500 。所有整数的范围在 -2^28 到 2^28 - 1 之间，最终结果不会超过 2^31 - 1 。

例如:

输入:

• A = [ 1, 2]
• B = [-2,-1]
• C = [-1, 2]
• D = [ 0, 2]

输出:

2

解释:

两个元组如下:

1. (0, 0, 0, 1) -> A[0] + B[0] + C[0] + D[1] = 1 + (-2) + (-1) + 2 = 0
2. (1, 1, 0, 0) -> A[1] + B[1] + C[0] + D[0] = 2 + (-1) + (-1) + 0 = 0

 

我的作答：

一到秋冬就变得不想学习呃呃呃...

先将四组数组分成两组，比如AB，CD，使用python里面的字典数据类型（c++可以用map）。

用字典的键储存每组的和（A[i]+B[j]），字典的值储存重复的次数（比如A[0]+B[1]=A[3]+B[2]=0）然后类似于有效字母异位词【代码随想录】刷题记录(13)-有效的字母异位词的思路，先获得AB里的所有和；

再遍历CD，获得目标值减CD的差值（target = 0-c-d），如果满足a+b+c+d=target，那么在字典的target键就是AB遍历时的a+b键，此时加上字典里储存的值，就是每一轮循环计算出的元组数量。

代码如下：

class Solution(object):
    def fourSumCount(self, nums1, nums2, nums3, nums4):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :type nums3: List[int]
        :type nums4: List[int]
        :rtype: int
        """
        mydict = {}
        count = 0
        for i in nums1:
            for j in nums2:
                if i+j not in mydict: #这一步搭配空字典很有用，适合不确定字典键数的时候
                    mydict[i+j]=0 #因为是对重复次数初始化，所以赋值0
                mydict[i+j] += 1
        
        for m in nums3:
            for n in nums4:
                target = 0-m-n
                if target in mydict:
                    count += mydict[target]
                
        return count

困扰我的地方在于如何创建的是一个空字典，但是又可以给指定键的值+1；我用的是上述代码的方法，不过我找到的参考代码也很有帮助，另一种简洁的写法，妙蛙。

参考代码（部分）：

hashmap = dict()
        for n1 in nums1:
            for n2 in nums2:
                hashmap[n1+n2] = hashmap.get(n1+n2, 0) + 1

dict.get()可以指定键值，学到了！