leetcode刷题之 给定两个数组,写一个方法来计算它们的交集。
给定两个数组,写一个方法来计算它们的交集。
例如:
给定 nums1 = [1, 2, 2, 1], nums2 = [2, 2], 返回 [2, 2].
注意:
输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
跟进:
如果给定的数组已经排好序呢?你将如何优化你的算法?
如果 nums1 的大小比 nums2 小很多,哪种方法更优?
如果nums2的元素存储在磁盘上,内存是有限的,你不能一次加载所有的元素到内存中,你该怎么办?
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
HashMap<Integer,Integer> cacheMap=new HashMap();
List<Integer> list=new ArrayList();
if(nums1.length==0||nums2.length==0) {
return new int[0];
}
for(Integer i:nums1){
int num=cacheMap.get(i)!=null?cacheMap.get(i):0;
cacheMap.put(i,num+1);
}
int index=0;
for(int j:nums2){
int num=cacheMap.get(j)!=null?cacheMap.get(j):0;
if(num>0){
list.add(j);
index++;
cacheMap.put(j,num-1);
}
}
int[] result = new int[list.size()];
int i = 0;
for (Integer e : list)
result[i++] = e;
return result;
}
}