349. 两个数组的交集(JavaScript)
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2] 输出: [2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4] 输出: [9,4]
说明:
- 输出结果中的每个元素一定是唯一的。
- 我们可以不考虑输出结果的顺序。
思路:
结果中的元素唯一,考虑用集合:set来保存。
/**
* @param {number[]} nums1
* @param {number[]} nums2
* @return {number[]}
*/
var intersection = function(nums1, nums2) {
var set = new Set();
if (!nums1 || !nums2) return [];
for (var n1 of nums1) {
if (nums2.indexOf(n1) !== -1) {
set.add(n1);
}
}
return Array.from(set);
};