Golang 四数相加 leetcode454 map哈希表

最新推荐文章于 2024-06-26 23:14:31 发布

鳄梨阿龙

最新推荐文章于 2024-06-26 23:14:31 发布

阅读量598

点赞数 9

分类专栏：个人 leetcode学习 Golang 文章标签：散列表数据结构算法

本文链接：https://blog.csdn.net/kingsill/article/details/135507375

版权

个人 leetcode学习 Golang 专栏收录该内容

21 篇文章 0 订阅

订阅专栏

本文介绍了如何通过使用两层for循环，将LeetCode454题中四数之和的原始O(N^4)时间复杂度降低到O(N^2)，通过记录并累加部分和出现次数来减少计算量。

摘要由CSDN通过智能技术生成

四数相加 leetcode454

本题如果直接进行四次for循环，则时间复杂度为O(N^4),超出运行时间限制。
因此我们这里使用两个分别的for循环进行遍历，则时间复杂度为O(N^2+N2).

/ 使用两遍for循环
func fourSumCount(nums1 []int, nums2 []int, nums3 []int, nums4 []int) int {

	index := 0

	record := map[int]int{}

	//首先对num3、nums4进行遍历，记录数据	考虑到可能每组有重复数字，每遍历到一次就+1
	for _, i1 := range nums3 {
		for _, i2 := range nums4 {
			record[i1+i2]++
		}

	}

	//然后再进行nums1、nums2的遍历
	for _, i1 := range nums1 {
		for _, i2 := range nums2 {
			//record对应数字是几，证明能得到这个数字的组合有几个，这里就加上对应的组合数
			if v, ok := record[-i1-i2]; ok {
				index = index + v
			}
		}

	}
	return index
}