go四数相加等于指定数算法

159 阅读1分钟

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

来源:力扣(LeetCode)
链接:leetcode-cn.com/problems/4s…
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
此题来源力扣
首先将四个数组分割为两两数组,前两个数组值相加,后两个数组相加,入股前两个数组相加和与后两个数组相加和正好为相反数,四个元素之和为0.
首先:将两数组的元素进行遍历相加,相加之和为map的索引。所指向的元素,就是出现的次数。

func foursumcount(A []int, B []int, C []int, D []int) int{
	des :=map[int]int{}
	for _,v:=range A{
		for _,w:=range B{
			des[v+w]++
		}
	}
}

再次遍历另两个数组,将两个数组的元素进行相加,取和的相反数,通过使用相反数在map中查找,如果没出现,所指向的数是0,如果出现过这个数的相反数,则所指向的数大于一。

func foursumcount(A []int, B []int, C []int, D []int) int{
	des :=map[int]int{}
	ans:=0
	for _,v:=range C{
		for _,w:=range D{
			ans +=des[-v-w]
		}
	}
}

最后将总数返回
全部代码

func fourSumCount(A []int, B []int, C []int, D []int) int {
	des := map[int]int{}
	ans:=0
	for _,v :=range A{//遍历两个数组,将两个数组的和作为一个索引,进行+1操作
		for _,w:=range B{
				des[v+w]++
		}
	}
	for _,v :=range C{//遍历另两个数组,如果这两个数组进行相加的和的相反数在map中不为1,则证明出现过
		for _,w:=range D{
			ans +=des[-v-w]
		}
	}
	return ans//返回总数
}