350. Intersection of Two Arrays II
Given two arrays, write a function to compute their intersection.
Example 1:
Input: nums1 = [1,2,2,1], nums2 = [2,2]
Output: [2,2]
Example 2:
Input: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
Output: [4,9]
1、解法一:先排序,后比较
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
ArrayList<Integer> al=new ArrayList<Integer>();
Arrays.sort(nums1);
Arrays.sort(nums2);
int i=0,j=0;
while(i<nums1.length&&j<nums2.length){
if(nums1[i]==nums2[j]){
al.add(nums1[i]);
i++;
j++;
}else if(nums1[i]<nums2[j]){
i++;
}else if(nums1[i]>nums2[j]){
j++;
}
}
int[] res=new int[al.size()];
for(int k=0;k<res.length;k++){
res[k]=al.get(k);
}
return res;
}
}
2、解法二:采用HashMap,先加入短的数组会快一点,但不如上一方法
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
List<Integer> al=(nums1.length<nums2.length)?helper(nums1,nums2):helper(nums2,nums1);
int[] res=new int[al.size()];
for(int k=0;k<res.length;k++){
res[k]=al.get(k);
}
return res;
}
public List<Integer> helper(int[] nums1, int[] nums2){
ArrayList<Integer> al=new ArrayList<Integer>();
HashMap<Integer,Integer> map=new HashMap<Integer,Integer>();
for(int i=0;i<nums1.length;i++){
if(map.containsKey(nums1[i])){
map.replace(nums1[i],map.get(nums1[i])+1);
}else{
map.put(nums1[i],1);
}
}
for(int j=0;j<nums2.length;j++){
if(map.containsKey(nums2[j])){
al.add(nums2[j]);
if(map.get(nums2[j])==1){
map.remove(nums2[j]);
}else{
map.replace(nums2[j],map.get(nums2[j])-1);
}
}
}
return al;
}
}