350. 两个数组的交集 II
1.题目描述
给你两个整数数组 nums1
和 nums2
,请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数,应与元素在两个数组中都出现的次数一致(如果出现次数不一致,则考虑取较小值)。可以不考虑输出结果的顺序。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2,2]
示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[4,9]
提示:
- 1 <= nums1.length, nums2.length <= 1000
- 0 <= nums1[i], nums2[i] <= 1000
进阶:
- 如果给定的数组已经排好序呢?你将如何优化你的算法?
- 如果 nums1 的大小比 nums2 小,哪种方法更优?
- 如果 nums2 的元素存储在磁盘上,内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/intersection-of-two-arrays-ii著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2.题解
2.1 排序+双指针
2.1.1 题解
首先讲一下题目,我一开始就理解错了。
这道题目对顺序是没有要求的,然后需要两个数组同时出现的数字,就和字符串匹配不一样。
因为对顺序没有要求,所以可以用排序,一个一个的去匹配,如果相同就放进去答案。
因为匹配前已经排好序了,所以如果其中有一个指针的数更小,就需要把这个指针向右移动,这样才能找到相同的数。代码的大致思路如下:
- 先将两个数组排序。
- 之后用两个指针指向两个数组的开头,如果偏小就向右移,如果相同则放入队列。
- 最后将队列转成数组返回。
2.1.2 Java代码
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
Arrays.sort(nums1);
Arrays.sort(nums2);
Queue<Integer> temp = new LinkedList<Integer>();
for(int i=0,j=0;i<nums1.length&&j<nums2.length;){
if(nums1[i]==nums2[j]){
temp.add(nums1[i]);
i++;
j++;
}
else if(nums1[i]>nums2[j]){
j++;
}
else{
i++;
}
}
int[] ans = new int[temp.size()];
for(int i=0;i<ans.length;i++){
ans[i] = temp.remove();
}
return ans;
}
}
2.2 哈希表
2.2.1 思路
哈希表的思路是用哈希表记录较短的数组中数字的出现次数,较长的数组去哈希表中找,如果哈希表中存在,就放入答案里。代码我用了两个函数来表达:
- 首先判断一下数组的大小,然后将较小的数组放入哈希表
- 之后用
compare
比较数字出现的次数,如果哈希表中存在且数字个数不为0,说明有重复的数字,放入答案。 - 最后返回数组的答案。
2.2.2 Java代码
class Solution {
HashMap<Integer,Integer> map = new HashMap<Integer,Integer>();
public int[] intersect(int[] nums1, int[] nums2) {
int m = nums1.length,n = nums2.length;
int []ans;
if(m<n){
makeMap(nums1);
ans = compare(nums2);
}
else{
makeMap(nums2);
ans = compare(nums1);
}
return ans;
}
public void makeMap(int[] nums){
int m = nums.length;
for(int i = 0;i < m ; i++){
if(map.containsKey(nums[i])){
int temp = map.get(nums[i])+1;
map.put(nums[i],temp);
}
else{
map.put(nums[i],1);
}
}
}
public int[] compare(int[] nums){
List<Integer> ans = new LinkedList<Integer>();
for(int i=0;i<nums.length;i++){
if(map.containsKey(nums[i])&&map.get(nums[i])>0){
ans.add(nums[i]);
int temp = map.get(nums[i])-1;
map.put(nums[i],temp);
}
}
int[] t = new int[ans.size()];
for(int i=0;i<t.length;i++){
t[i] = ans.get(i);
}
return t;
}
}