给定两个数组,编写一个函数来计算它们的交集。
示例 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 的元素存储在磁盘上,内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/intersection-of-two-arrays-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
func intersect(nums1 []int, nums2 []int) []int {
m0 := map[int] int{}
for _, i := range nums1 {
m0[i] += 1
}
k := 0
for _, j := range nums2 {
if m0[j] > 0 {
m0[j] -= 1
nums2[k] = j;
k++
}
}
return nums2[0:k]
}
func intersect(nums1 []int, nums2 []int) []int {
i, j, k := 0, 0, 0
sort.Ints(nums1)
sort.Ints(nums2)
numSize1 := len(nums1)
numSize2 := len(nums2)
for i < numSize1 && j < numSize2 {
if nums1[i] > nums2[j] {
j++
} else if nums1[i] < nums2[j] {
i++
}else {
nums1[k] = nums1[i]
i++
j++
k++
}
}
return nums1[0:k]
}
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
int numSize1 = nums1.size(), numSize2 = nums2.size();
unordered_map <int, int> m;
for (int i = 0; i < numSize1; i++) {
m[nums1[i]] += 1;
}
int k = 0;
vector<int> ans;
for (int i = 0; i < numSize2; i++) {
if (m.count(nums2[i])) {
m[nums2[i]] -= 1;
ans.push_back(nums2[i]);
k++;
if (m[nums2[i]] == 0) {
m.erase(nums2[i]);
}
}
}
return ans;
}
};