leetcode 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]

中文理解:给定两个数组,返回两个数组中所有重复的元素。

解题思路:如果两个数组都是排序好的话,直接采用类似于有序链表合并的思路解决,不过本题数组是乱序的,所以可以先把两个数组排序,也可以使用hashmap存放每个数字出现的次数,然后得出两个hashmap重合出现的key值,然后取value最小的值,把最小出现次数的key值放入数组,最后返回。

代码(java):

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        HashMap map1=new HashMap();
        HashMap map2=new HashMap();
        for(int val:nums1){
            if(map1.keySet().contains(val)){
                map1.put(val,map1.get(val)+1);
            }
            else{
                map1.put(val,1);
            }
        }
        for(int val:nums2){
            if(map2.keySet().contains(val)){
                map2.put(val,map2.get(val)+1);
            }
            else{
                map2.put(val,1);
            }
        }
        Set com=map1.keySet();
        com.retainAll(map2.keySet());
        int len=0;
        for(int val:com){
            len+=Math.min(map1.get(val),map2.get(val));
        }
        int []res=new int[len];
        int i=0;
        for(int val:com){
            int count=Math.min(map1.get(val),map2.get(val));
            while(count>0){
                res[i]=val;
                i++;
                count--;
            }
        }
        return res;
    }
}

 

你可能感兴趣的:(leetcode)