leetcode笔记_集合和映射的使用及比较 (集合的元素去重，映射的元素频次记录)

最新推荐文章于 2022-05-08 13:38:53 发布

Crownt

最新推荐文章于 2022-05-08 13:38:53 发布

阅读量200

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/jt102605/article/details/84191886

版权

leetcode 专栏收录该内容

17 篇文章 0 订阅

订阅专栏

349. 两个数组的交集

给定两个数组，编写一个函数来计算它们的交集。

示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]
说明:

输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。

思路：利用集合的去重特点

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        
        //通过set对nums1去重
        TreeSet<Integer> set = new TreeSet<>();
        for(int num: nums1){
            set.add(num);
        }
        
        //判断set中是否包含nums2中的元素，若包含则记录下来，并在set中删除此元素，防止重复记录nums2中的重复元素
        ArrayList<Integer> list = new ArrayList<>();
        for(int num: nums2){
            if(set.contains(num)){
                list.add(num);
                set.remove(num);  //注意，这里的移除操作，是为了防止nums2中的重复元素被包含进来
            }
        }
        
        //题目要求返回数组
        int[] arr = new int[list.size()];
        for(int i=0; i<arr.length; i++){
            arr[i] = list.get(i);
        }
              
        return arr;
    }
}

350. 两个数组的交集 II

给定两个数组，编写一个函数来计算它们的交集。

示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明：

输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。

思路：使用映射实现元素频次的记录

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        
        //通过Map记录nums1中出现的元素，以及每个元素出现的次数，K记录出现的元素，V记录该元素出现的次数
        TreeMap<Integer, Integer> map = new TreeMap<>();
        for(int num:nums1){
            if(!map.containsKey(num)){
                map.put(num, 1);
            }else{
                map.put(num, map.get(num)+1);
            }
        }
        
        
        //判断map中是否包含nums2的元素，若包含，则记录此元素，并将对应元素的次数减一，减为0时，则从map中删除此元素
        ArrayList<Integer> list = new ArrayList<>();
        for(int num:nums2){
            if(map.containsKey(num)){
                list.add(num);
                map.put(num, map.get(num)-1);
                if(map.get(num) == 0){
                    map.remove(num);
                }
            }
        }
        
        //题目要求返回数组
        int[] arr = new int[list.size()];
        for(int i=0; i<arr.length; i++){
            arr[i] = list.get(i);
        }
        
        return arr;
    }
}