350. 两个数组的交集 II
https://leetcode-cn.com/problems/intersection-of-two-arrays-ii/
给定两个数组,编写一个函数来计算它们的交集。
- 示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2,2]
- 示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[4,9]
解法一:HashMap
- 使用Map统计nums1数字的个数
- 遍历nums2,查找当前数字是否在Map中
- 若在且出现个数大于0,说明这个数是交集中的一个元素
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
if (nums1.length * nums2.length == 0) return new int[]{};
Map<Integer, Integer> map = new HashMap<>();
for (int i : nums1) {
map.put(i, map.getOrDefault(i, 0) + 1);
}
int index = 0;
for (int i : nums2) {
if (map.containsKey(i)) {
int cnt = map.get(i);
if (cnt > 0) {
nums2[index++] = i;
map.put(i, cnt - 1);
}
}
}
return Arrays.copyOf(nums2, index);
}
}
- 时间复杂度:
O(M + N)
- 空间复杂度:
O(M)
或者O(N)
解法二:排序+双指针
- 分别对nums1、nums2进行排序
- 使用指针i,j分别对nums1、nums2遍历
- nums[i]==nums[j]时,为一个答案
- nums[i] > nums[j]时,j++
- nums[i] < nums[j]时,i++
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
if (nums1.length * nums2.length == 0) return new int[]{};
Arrays.sort(nums1); Arrays.sort(nums2);
int i = 0, j = 0, k = 0;
int[] ans = new int[nums1.length];
while (i < nums1.length && j < nums2.length) {
if (nums1[i] == nums2[j]) {
ans[k++] = nums1[i++];
j++;
} else if (nums1[i] < nums2[j]) {
i++;
} else {
j++;
}
}
return Arrays.copyOf(ans, k);
}
}
- 时间复杂度:O(MlogM + NlogN)
- 空间复杂度:O(M) 或者 O(N)