【Leetcode】454. 四数相加 II

题目链接

【Leetcode】454. 四数相加 II

代码

func fourSumCount(nums1 []int, nums2 []int, nums3 []int, nums4 []int) int {
   
	ans := 0
	// 思路，分组 + hash 每个数组长度最多200，如果暴力四层循环，会超时，因此使用hash表把两层循环优化成一层
	// 枚举nums1,nums2，统计v1+v2出现的次数
	hash := make(map[int]int, 0)
	for _, v1 := range nums1 {
   
		for _, v2 := range nums2 {
   
			hash[v1+v2]++
		}
	}
	// 枚举nums3,nums4,统计v1+v2==-(v3+v4)出现的次数
	for _, v3 := range nums3 {
   
		for _, v4 := range nums4 {
   
			if cnt, ok := hash[-(v3 + v4)]; ok {
   
				ans += cnt
			}
		}
	}
	return ans
}