Given two arrays, write a function to compute their intersection.
Example:
Givennums1=[1, 2, 2, 1],nums2=[2, 2], return[2, 2].
Note:
- Each element in the result should appear as many times as it shows in both arrays.
- The result can be in any order.
Follow up:
- What if the given array is already sorted? How would you optimize your algorithm?
- What if nums1 's size is small compared to nums2 's size? Which algorithm is better?
- What if elements of nums2 are stored on disk, and the memory is limited such that you cannot load all elements into the memory at once?
- time o(n) space O (n)
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
if (nums1 == null || nums2 == null) {
return new int[0];
}
List<Integer> res = new ArrayList<>();
HashMap<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums1.length; i++) {
if (map.containsKey(nums1[i])) {
map.put(nums1[i], map.get(nums1[i]) + 1);
}else{
map.put(nums1[i], 1);
}
}
for (int i = 0; i < nums2.length; i++) {
if (map.containsKey(nums2[i])) {
if (map.get(nums2[i]) > 0) {
res.add(nums2[i]);
map.put(nums2[i], map.get(nums2[i]) - 1);
}
}
}
int[] ans = new int[res.size()];
int index = 0;
for (Integer i : res) {
ans[index++] = i;
}
return ans;
}
}