349. 两个数组的交集 golang

两个数组交集算法

最新推荐文章于 2024-08-19 04:59:38 发布

原创最新推荐文章于 2024-08-19 04:59:38 发布 · 707 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

152 篇文章

订阅专栏

本文介绍了一种计算两个数组交集的算法实现，通过使用哈希表存储元素，避免了重复输出并确保了输出元素的唯一性。示例中展示了如何处理不同输入情况，如[1,2,2,1]与[2,2]的交集为[2]，而[4,9,5]与[9,4,9,8,4]的交集为[9,4]。

349. 两个数组的交集

给定两个数组，编写一个函数来计算它们的交集。

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]
示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]
说明:

输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。

Code

func intersection(nums1 []int, nums2 []int) []int {
	set := make(map[int]bool)
	res := make([]int, 0)

	for _, v1 := range nums1 {
		set[v1] = true
	}

	// [1,2,2,1]
	// [2,2]

	// 用hash的value来进行true， false判断。值得借鉴
	for _, v2 := range nums2 {
		if true_or_false, ok := set[v2]; ok && true_or_false {  //nums2里面包含nums1里的元素
			res = append(res, v2)
			set[v2] = false             //防止重复输出
		}
	}
	return res
}