LeetCode 349：两个数组的交集 Intersection of Two Arrays
LeetCode 349：两个数组的交集 Intersection of Two Arrays
爱写bug
发布于 2019-09-29 17:05:32
发布于 2019-09-29 17:05:32
文章被收录于专栏：爱写Bug
题目：
给定两个数组，编写一个函数来计算它们的交集。
Given two arrays, write a function to compute their intersection.
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]
示例 2:
输入: 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:
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语言里有交集、并集、差集的概念：
A 与 B交集：A & B
A 与 B并集：A | B
A 与 B差集：A - B
所以一行代码即可解题：
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 删除
评论
登录后参与评论
推荐阅读
目录
腾讯云开发者
