代码随想录算法训练营第七天454.四数相加

代码随想录算法训练营第七天|454.四数相加

学习文档：卡哥讲解

题目简介：

​ 给定四个相同长度的列表，要求返回多少个元组（i,j,k,l）其中A[i]+B[j]+C[k]+D[l]=0.

哈希表

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

第一下不会写，

1. 先列举出字典中的键和值。键就是两数之和，分别遍历。值就是出现过加1.
2. 再遍历另外两个数组，因为A[i]+B[j]+C[k]+D[l]=0. 所以 C[k]+D[l]=-（A[i]+B[j]）如果此和在另外两个数组中，则表示符合条件元素存在。