349 Intersection of Two Arrays

最新推荐文章于 2022-10-31 23:18:53 发布

原创最新推荐文章于 2022-10-31 23:18:53 发布 · 63 阅读

CC 4.0 BY-SA版权

博客围绕计算两个数组交集的问题展开。介绍了解题思路，即先对数组去重再取交集。给出两种解法，一是用基于平衡二叉树的 TreeSet，时间复杂度 O(NlogN)，空间复杂度 O(N)；二是用基于哈希表的 HashSet，时间复杂度 O(len(nums1)+len(nums2))，空间复杂度 O(len(nums1))。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

Given two arrays, write a function to compute their intersection.

Example 1:

Input: nums1 = [1,2,2,1], nums2 = [2,2]
Output: [2]

Example 2:

Input: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
Output: [9,4]

Note:
Each element in the result must be unique.
The result can be in any order.

解法思路

先对 nums1 去重，然后在和 nums2 取交集；

解法实现（一）使用基于平衡二叉树实现的 TreeSet

对于 nums1 来说，和 nums2 重复的元素要“移动”到交集中；

关键词

Set TreeSet

时间复杂度

O(NlogN)；
标准库中的 Set 是基于平衡的二叉树实现的，其时间复杂度为 O(logN)；

空间复杂度

O(N)；

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

        ArrayList<Integer> list = new ArrayList<>();
        for (int num : nums2) {
            if (set.contains(num)) {
                list.add(num);
                set.remove(num);
            }
        }

        int[] res = new int[list.size()];
        for (int i = 0; i < list.size(); i++) {
            res[i] = list.get(i);
        }
        return res;
    }
    
}

解法实现（二）使用基于哈希表实现的 HashSet

时间复杂度

O(len(nums1)+len(nums2))；

空间复杂度

O(len(nums1))；

关键词

Set HashSet

import java.util.HashSet;

// 349. Intersection of Two Arrays
// https://leetcode.com/problems/intersection-of-two-arrays/description/
// 时间复杂度: O(len(nums1)+len(nums2))
// 空间复杂度: O(len(nums1))
public class Solution349 {

    public int[] intersection(int[] nums1, int[] nums2) {

        HashSet<Integer> record = new HashSet<Integer>();
        for(int num: nums1)
            record.add(num);

        HashSet<Integer> resultSet = new HashSet<Integer>();
        for(int num: nums2)
            if(record.contains(num))
                resultSet.add(num);

        int[] res = new int[resultSet.size()];
        int index = 0;
        for(Integer num: resultSet)
            res[index++] = num;

        return res;
    }

    private static void printArr(int[] arr){
        for(int e: arr)
            System.out.print(e + " ");
        System.out.println();
    }

    public static void main(String[] args) {

        int[] nums1 = {1, 2, 2, 1};
        int[] nums2 = {2, 2};
        int[] res = (new Solution349()).intersection(nums1, nums2);
        printArr(res);
    }
}