目录
- 方法一:哈希表
- 方法二:排序 + 双指针
题目来源
350. 两个数组的交集 II
方法一:哈希表
由于同一个数字在两个数组中都可能出现多次,因此需要用哈希表存储每个数字出现的次数。对于一个数字,其在交集中出现的次数等于该数字在两个数组中出现次数的最小值。
首先遍历第一个数组,并在哈希表中记录第一个数组中的每个数字以及对应出现的次数,然后遍历第二个数组,对于第二个数组中的每个数字,如果在哈希表中存在这个数字,则将该数字添加到答案,并减少哈希表中该数字出现的次数。
为了降低空间复杂度,首先遍历较短的数组并在哈希表中记录每个数字以及对应出现的次数,然后遍历较长的数组得到交集。
时间复杂度:O(m+n)
空间复杂度:O(min(m,n))
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
// 挑选出两个数组找个短的那个
if (nums1.length > nums2.length) {
return intersect(nums2, nums1);
}
HashMap<Integer, Integer> map = new HashMap<>();
for (int num : nums1) {
// map.getOrDefault : 存在这个数就返回,不存在就返回默认值
int count = map.getOrDefault(num, 0) + 1;
map.put(num, count);
}
// 开辟一块内存空间用来存放两个数组的交集
int[] intersection = new int[nums1.length];
int index = 0;
for (int num : nums2) {
// num1中不存在这个数就在map中添加num=0
int count = map.getOrDefault(num, 0);
if (count > 0) {
intersection[index++] = num;
count--;
// 如果还大于0
if (count > 0) {
// 再次添加进去,覆盖之前那个key
map.put(num, count);
} else {
// 不大于0移除这个数
map.remove(num);
}
}
}
return Arrays.copyOfRange(intersection,0,index);
}
}
map.getOrDefault(Object key, V defaultValue);
①map中存在key,value返回key对应的value即可。
②map中不存在key,value则返回defaultValue(默认值)。
方法二:排序 + 双指针
创建一个指针 i 指向 nums1 数组首位,指针 j 指向nums2 数组首位。
创建一个临时栈,用于存放结果集。
开始比较指针 i 和指针 j 的值大小,若两个值不等,则数字小的指针,往右移一位。
若指针 i 和指针 j 的值相等,则将交集压入栈。
若 nums 或 nums2 有一方遍历结束,代表另一方的剩余值,都是唯一存在,且不会与之产生交集的。
时间复杂度:O(mlogm+nlogn)
空间复杂度:O(min(m,n))
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
Arrays.sort(nums1);
Arrays.sort(nums2);
int[] arr = new int[Math.min(nums1.length, nums2.length)];
int l = 0;
int j = 0;
int index = 0;
while (l < nums1.length && j < nums2.length){
if (nums1[l] < nums2[j]){
l++;
}else if (nums1[l] > nums2[j]){
j++;
}else {
arr[index] = nums1[l];
l++;
j++;
index++;
}
}
return Arrays.copyOfRange(arr,0,index);
}
}