leetcode学习笔记45

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;
    }
}