【算法刷题】3:两个数组的交集

给定两个数组,写一个方法来计算他们的交集。 示例:数组

给定 nums1 = [1, 2, 2, 1], nums2 = [2, 2], 返回 [2, 2].
复制代码

注意:bash

  • 输出结果中每一个元素出现的次数,应与元素在两个数组中出现的次数一致。
  • 咱们能够不考虑输出结果的顺序。

分析

记住要记录数字及其出现次数,须要对这两个数组都遍历一遍,前后在容器记录nums1中出现的数字及其次数,在遍历查询容器中是否存在nums2中的数字,好在这一次不须要再末尾遍历输出,也就是说咱们只须要一个基础的键值对容器便可。ui

代码

private static int[] interserct(int[] nums1, int[] nums2) {
        // 用来暂存结果,保存数组的长度以及各元素的值
        List<Integer> tmp = new ArrayList<>();
        HashMap<Integer, Integer> map = new HashMap<>();
        // 保存nums1中元素的个数
        for (int i = 0; i < nums1.length; i++) {
            Integer temp = map.get(nums1[i]);
            map.put(nums1[i], (temp == null ? 1 : temp + 1));
        }

        // 遍历nums2,查询
        for (int i = 0; i < nums2.length; i++) {
            if (map.containsKey(nums2[i]) && map.get(nums2[i]) != 0) {
                //存放该元素,且次数>0,则次数-1
                tmp.add(nums2[i]);
                map.put(nums2[i], map.get(nums2[i]) - 1);
            }
        }
        int[] result = new int[tmp.size()];
        int i = 0;
        for (Integer in : tmp) {
            result[i++] = in;
        }
        return result;
    }
复制代码
相关文章
相关标签/搜索