leetcode-2215. 找出两数组的不同

41 阅读2分钟

题目要求:

给你两个下标从 0 开始的整数数组 nums1 和 nums2 ,请你返回一个长度为 2 的列表 answer ,其中:

answer[0] 是 nums1 中所有 不 存在于 nums2 中的 不同 整数组成的列表。 answer[1] 是 nums2 中所有 不 存在于 nums1 中的 不同 整数组成的列表。 注意:列表中的整数可以按 任意 顺序返回。

示例 1: 输入:nums1 = [1,2,3], nums2 = [2,4,6] 输出:[[1,3],[4,6]] 解释: 对于 nums1 ,nums1[1] = 2 出现在 nums2 中下标 0 处,然而 nums1[0] = 1 和 nums1[2] = 3 没有出现在 nums2 中。因此,answer[0] = [1,3]。 对于 nums2 ,nums2[0] = 2 出现在 nums1 中下标 1 处,然而 nums2[1] = 4 和 nums2[2] = 6 没有出现在 nums2 中。因此,answer[1] = [4,6]。

示例 2: 输入:nums1 = [1,2,3,3], nums2 = [1,1,2,2] 输出:[[3],[]] 解释: 对于 nums1 ,nums1[2] 和 nums1[3] 没有出现在 nums2 中。由于 nums1[2] == nums1[3] ,二者的值只需要在 answer[0] 中出现一次,故 answer[0] = [3]。 nums2 中的每个整数都在 nums1 中出现,因此,answer[1] = [] 。

提示: 1 <= nums1.length, nums2.length <= 1000 -1000 <= nums1[i], nums2[i] <= 1000

链接leetcode.cn/problems/fi…

解题思路

使用集合来做,因为对于重复的数字需要去重。go语言的集合比较麻烦,需要使用map来实现。

代码如下

func findDifference(nums1 []int, nums2 []int) [][]int {
	// 使用集合来实现
	set1 := make(map[int]bool, len(nums1))
	for _, v := range nums1 {
		set1[v] = true
	}
	set2 := make(map[int]bool, len(nums2))
	for _, v := range nums2 {
		set2[v] = true
	}
	result1 := make([]int, 0)
	result2 := make([]int, 0)
	// 遍历set查询不在集合中的元素
	for v := range set1 {
		if !set2[v] {
			result1 = append(result1, v)
		}
	}
	for v := range set2 {
		if !set1[v] {
			result2 = append(result2, v)
		}
	}
	return [][]int{result1, result2}
}