LeetCode Top Interview Questions 350. Intersection of Two Arrays II (Java版; Easy)

welcome to my blog

LeetCode Top Interview Questions 350. Intersection of Two Arrays II (Java版; Easy)

题目描述

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

第一次做; 使用双指针, 这个方法需要先对两个数组进行排序, 但是处理思路非常好; 双指针: 小的走, 相同的一起走; 这个方法也可以解决超大规模数据的问题, 不过得先将超大规模的数据排序, 这属于大数问题排序…去学习一下

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        if(nums1==null || nums1.length==0 || nums2==null || nums2.length==0)
            return new int[]{};
        //
        //先对两个数组进行排序
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        List<Integer> list = new ArrayList<>();
        int i=0, j=0;
        while(i<nums1.length && j<nums2.length){
            int cur1 = nums1[i];
            int cur2 = nums2[j];
            if(cur1<cur2)
                i++;
            else if(cur2<cur1)
                j++;
            else{
                list.add(cur1);
                i++;
                j++;
            }
        }
        //java8新特性
        return list.stream().mapToInt(k -> k).toArray();
    }
}

第一次做; 只用一个哈希表, 先统计nums1中各个元素出现的次数, 然后遍历nums2, 出现相同的元素加入到list中, 并且将该元素的出现次数减一; 最后使用java8新特性将ArrayList转成int数组

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        if(nums1==null || nums1.length==0 || nums2==null || nums2.length==0)
            return new int[]{};
        //
        HashMap<Integer, Integer> map = new HashMap<>();
        List<Integer> list = new ArrayList<>();
        for(int i=0; i<nums1.length; i++){
            int cur = nums1[i];
            map.put(cur, map.getOrDefault(cur,0)+1);
        }
        for(int i=0; i<nums2.length; i++){
            int cur = nums2[i];
            if(map.getOrDefault(cur,0)>0){
                list.add(cur);
                map.put(cur, map.get(cur)-1);
            }
        }
        //java8 新特性, 从ArrayList转成整型数组
        return list.stream().mapToInt(k -> k).toArray();
    }
}

第一次做; 最笨的方法, 使用两个哈希表, 主要是熟悉熟悉如何遍历哈希表; 哈希表的key是数组中的元素, value是元素的索引也可以是元素的出现次数

//这不是常规的集合, 因为含有重复元素; 那么该怎么解决重复元素问题? 用两个哈希表, key是元素值, value是个list, 存储满足条件的索引
class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        if(nums1==null || nums1.length==0 || nums2==null || nums2.length==0)
            return new int[]{};
        //
        HashMap<Integer, List<Integer>> map1 = new HashMap<>();
        HashMap<Integer, List<Integer>> map2 = new HashMap<>();
        //统计nums1每个元素及其对应的索引
        for(int i=0; i<nums1.length; i++){
            int cur = nums1[i];
            if(!map1.containsKey(cur))
                map1.put(cur, new ArrayList<Integer>());
            map1.get(cur).add(i);
        }
        //统计nums2每个元素及其对应的索引
        for(int i=0; i<nums2.length; i++){
            int cur = nums2[i];
            if(!map2.containsKey(cur))
                map2.put(cur, new ArrayList<Integer>());
            map2.get(cur).add(i);
        }
        //统计相同元素的个数
        int count = 0;
        for(Map.Entry<Integer, List<Integer>> e : map1.entrySet()){
            int key = e.getKey();
            if(map2.containsKey(key)){
                count += Math.min(e.getValue().size(), map2.get(key).size());
            }
        }
        //将相同元素存入数组
        int[] res = new int[count];
        int i = 0;
        for(Map.Entry<Integer, List<Integer>> e : map1.entrySet()){
            int key = e.getKey();
            if(map2.containsKey(key)){
                int num = Math.min(e.getValue().size(), map2.get(key).size());
                for(int j=0; j<num; j++){
                    res[i++] = key;
                }
            }
        }
        return res;
    }
}