栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > C/C++/C#

【leetcode454】四数相加--go语言实现

C/C++/C# 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

【leetcode454】四数相加--go语言实现

题目描述:

给你四个整数数组 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

思路分析:

定义变量res,用来统计 a+b+c+d = 0 出现的次数。 遍历前两个数组a和b,把元素之和,和出现的次数,放到map中。
然后遍历后面两个数组c和d,如果 0-(c+d)
在map中出现过的话,就用count把map中key对应的value也就是出现次数统计出来,累加到res上, 最后返回统计值 res就可以了

具体代码
func fourSumCount(nums1 []int, nums2 []int, nums3 []int, nums4 []int) int {
	hashmap1:=map[int]int{}
	res:=0
	for _,a:=range nums1{
		for _,b:=range nums2{
			hashmap1[a+b]++
		}
	}
	for _,c:=range nums1{
		for _,d:=range nums2{
			if count,ok:=hashmap1[0-c-d];ok{
				res+=count
			}
		}
	}
	return res
}
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/938284.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号