春招刷题 - 349. 两个数组的交集

151 阅读1分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

一、题目描述:

349. 两个数组的交集 - 力扣(LeetCode) (leetcode-cn.com)

给定两个数组 nums1 和 nums2 ,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序

示例 1:

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

示例 2:

输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
解释:[4,9] 也是可通过的

提示:

  • 1 <= nums1.length, nums2.length <= 1000
  • 0 <= nums1[i], nums2[i] <= 1000

二、思路分析:

可以用到集合来解决,集合一般是没有重复元素的,那么就可以先对数组1去重,去重后再遍历数组2,如果在集合中存在的,说明是数组1和数组2都有的元素,就加入结果数组中,为了避免数组2中重复这个元素,需要把集合中的删除,第二次再遇到时集合里面没有了,就不会重复添加到结果数组中。

三、AC 代码:

import java.util.ArrayList;
import java.util.TreeSet;

public class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        TreeSet<Integer> set = new TreeSet<>();
        for (int i:nums1) {
            set.add(i);
        }

        ArrayList<Integer> arrayList = new ArrayList<>();
        for (int i:nums2) {
            if(set.contains(i)){
                arrayList.add(i);
                set.remove(i);
            }
        }
        
        int[] res = new int[arrayList.size()];
        
        for (int i=0;i<arrayList.size();i++){
            res[i] = arrayList.get(i);
        }
        
        return res;
    }
}

四、总结:

写完开开心心过周末了~

范文参考

C++ 简单哈希 - 两个数组的交集 - 力扣(LeetCode) (leetcode-cn.com)