361 两个数的交集 + 双指针+ 字典

题目描述

给你两个整数数组 nums1 和 nums2 ，请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数，应与元素在两个数组中都出现的次数一致（如果出现次数不一致，则考虑取较小值）。可以不考虑输出结果的顺序。

示例 1：

输入：nums1 = [1,2,2,1], nums2 = [2,2]
输出：[2,2]
示例 2:

输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[4,9]

思路

创建一个map
将数组1里面的所有数都放进去
key为 数组的元素
value 为 其出现次数

遍历数组2
如果 数组内的元素 在 map中存在的话，
就

代码

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        //create a dict with key-value
        HashMap<Integer,Integer> map = new HashMap<>();
        // create an list
        ArrayList<Integer> list = new ArrayList<>();

        // iterate through the array
        // add the term in array to the map 
        // the key is the term value 
        // the value is the Number of occurrences of terms
        for(int i: nums1){
            map.put(i, map.getOrDefault(i,0) + 1); 
            // getOrDefault(Object key, V defaultValue) 如果给定的 key 在映射关系中找不到，则返回指定的默认值。
            // getOrDefalut means that if cannot find the key, return 0, if not return the value of the key in the dic
        }

        for(int i: nums2){
            if(map.getOrDefault(i,0) > 0){
                list.add(i);
                map.put(i,map.get(i) - 1);
            }
        }

        // 把list转为数组
        int[] res = new int[list.size()];
        for(int i = 0; i < list.size(); i++){
            res[i] = list.get(i);
        }

        return res;
    }
}

思路2

使用双指针

第一个指针从第一个数组
第二个指针从第二个数组

如果相同，就把这个元素添加到新的ArrayList中
否则，就移动小的那个元素的指针，直至某个指针走到尽头

代码2

class Solution {
        public int[] intersect(int[] nums1, int[] nums2) {
        // sort the two array
        Arrays.sort(nums1);
        Arrays.sort(nums2);

        int p1 = 0;
        int p2 = 0;
        ArrayList<Integer> List = new ArrayList<>();

        while( p1 < nums1.length && p2 < nums2.length){

            if (nums1[p1] < nums2[p2]){
                p1++;
            } else if (nums2[p2] < nums1[p1]){
                p2++;
            } else {
                List.add(nums1[p1]);
                p1++;
                p2++;
            }
        }

        // transfer the List into array
        int[] temp = new int[List.size()];
        for (int i = 0; i < List.size(); i++){
            temp[i] = List.get(i);
        }

        return temp;
    }
}

Reference

链接：https://leetcode.cn/leetbook/read/top-interview-questions-easy/x2y0c2/