548.Intersection of Two Arrays II-两数组的交 II（容易题）

最新推荐文章于 2021-11-05 07:47:01 发布

原创最新推荐文章于 2021-11-05 07:47:01 发布 · 272 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LintCode #hashmap

LintCode笔记专栏收录该内容

259 篇文章

订阅专栏

本文介绍了一种计算两个整数数组交集的方法，确保每个元素出现的次数与其在原始数组中的出现次数一致。通过使用HashMap来跟踪元素及其频率，实现高效查找并更新结果。

两数组的交 II

题目

计算两个数组的交

注意事项
每个元素出现次数得和在数组里一样
答案可以以任意顺序给出
样例

nums1 = [1, 2, 2, 1], nums2 = [2, 2], 返回 [2, 2].
题解

先对nums1遍历并将nums1[i]和nums1[i]出现次数存入HashMap，再对nums2进行遍历，如果任意nums2[i]在HashMap中存在且Value大于0则标记为相交数，并将Value计数-1。

public class Solution {
    /**
     * @param nums1 an integer array
     * @param nums2 an integer array
     * @return an integer array
     */
    public int[] intersection(int[] nums1, int[] nums2) {
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
        List<Integer> results = new ArrayList<Integer>();

        for(int i = 0; i < nums1.length; ++i) 
        {
            map.put(nums1[i], map.containsKey(nums1[i])?(map.get(nums1[i]) + 1):1); 
        }

        for (int i = 0; i < nums2.length; ++i)
        {
            if (map.containsKey(nums2[i]) && map.get(nums2[i]) > 0) 
            {
                results.add(nums2[i]);
                map.put(nums2[i], map.get(nums2[i]) - 1); 
            }
        }

        int result[] = new int[results.size()];
        for(int i = 0; i < results.size(); ++i)
        {
            result[i] = results.get(i);
        }

        return result;
    }
}