前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 349:两个数组的交集 Intersection of Two Arrays

LeetCode 349:两个数组的交集 Intersection of Two Arrays

作者头像
爱写bug
发布2019-09-29 17:05:32
3930
发布2019-09-29 17:05:32
举报
文章被收录于专栏:爱写Bug

题目:

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

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

示例 1:

代码语言:javascript
复制
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]

示例 2:

代码语言:javascript
复制
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]

说明:

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

Note:

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

解题思路:

三次遍历通用解题:

  • 第一次 nums1 加入 set1 ()集合,此时 set1() 内元素为去重后的 nums1 集合。
  • 第二次 将 nums2 中的同时存在于 set1 的元素加入到 set 2() 集合,此时 set2() 集合即为两个数组的交集。
  • 第三次 将 set2() 集合中的元素提取出作为整型数组返回。

代码:

Java:

代码语言:javascript
复制
class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        Set<Integer> set1 = new HashSet<>();
        Set<Integer> set2 = new HashSet<>();
        for (int num : nums1) //去重
            set1.add(num);
        for (int num : nums2) {
            if (set1.contains(num)) set2.add(num); //依然用 set() 集合存交集元素以防止元素重复
        }
        int[] res = new int[set2.size()];
        int i = 0;
        for (int num : set2)//转为 int[]
            res[i++] = num;
        return res;
    }
}

Python:

在Py语言里有交集、并集、差集的概念:

代码语言:javascript
复制
A 与 B交集:A & B
A 与 B并集:A | B
A 与 B差集:A - B

所以一行代码即可解题:

代码语言:javascript
复制
class Solution:
    def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
        # nums1、nums2转为set集合并取交集,并将交集元素转为list数组返回
        return list(set(nums1) & set(nums2))
本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2019-09-27,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 爱写Bug 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 题目:
  • 解题思路:
  • 代码:
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档