给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明:
输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
进阶:
如果给定的数组已经排好序呢?你将如何优化你的算法?
如果 nums1 的大小比 nums2 小很多,哪种方法更优?
如果 nums2 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
分析:
这道题意思很明显,找出两个数组都存在的元素。这道题本身不难,就算是遍历穷举的方法也能做出来,但肯定会导致时间复杂度高,不够优雅。
func intersect(nums1 []int, nums2 []int) []int {
if nums1 == nil || nums2 == nil {
return []int{}
}
(nums1)
(nums2)
var x = 0
var y = 0
var result []int
for {
if x < len(nums1) && y < len(nums2) {
if nums1[x] == nums2[y] {
result = append(result, nums1[x])
x++;
y++;
} else if nums1[x] > nums2[y] {
y++
} else {
x++
}
} else {
break
}
}
return result
}
我们可以先递增排序,这样可以利用顺序来过滤掉已经不可能出现交集的元素,再定义两个脚标变量,这两个变量都从0开始,向后遍历,如果两个数组的值相同,存下这个值,两个脚标同时加1;如果两个数组的值不同,那么值较小的数组脚标值加1,另一个脚标值等待……直到遍历完短的那个数组……