一.题目
给定两个数组 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
二.题目解析+思路
暴力枚举:
首先遍历第一个数组
nums1
,然后对于nums1
中的每个元素,再遍历第二个数组nums2
。如果找到相同的元素,并且这个元素还没有被添加到结果数组result
中,就将其添加到result
中。时间复杂度是:O(n×m)
set优化:
利用set优秀的自动去重功能简化做题步骤。
哈希表:
思路和set差不多,但效率更高。
三.代码示例
暴力:
class Solution
{
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2)
{
vector<int> result;
for (int i = 0; i < nums1.size(); ++i)
{
for (int j = 0; j < nums2.size(); ++j)
{
if (nums1[i] == nums2[j])
{
// 检查是否已经添加到结果中,避免重复
bool found = false;
for (int k = 0; k < result.size(); ++k)
{
if (result[k] == nums1[i])
{
found = true;
break;
}
}
if (!found)
{
result.push_back(nums1[i]);
}
}
}
}
return result;
}
};
首先遍历第一个数组 nums1
,然后对于 nums1
中的每个元素,再遍历第二个数组 nums2
。如果找到相同的元素,并且这个元素还没有被添加到结果数组 result
中,就将其添加到 result
中。
这种方法的时间复杂度是 O(n×m),其中 n 和 m 分别是两个数组的长度。
set:
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2)
{
set<int> a1(nums1.begin(),nums1.end());
set<int> a2(nums2.begin(),nums2.end());
vector<int> v;
for(auto x:a1)
{
if(a2.count(x))
{
v.push_back(x);
}
}
return v;
}
};
代码讲解:
-
将数组转换为
使用set
:set<int>
将nums1
和nums2
分别转换为两个set
,命名为a1
和a2
。这一步自动去除了数组中的重复元素,并允许快速查找。 -
初始化结果向量:
创建一个vector<int>
类型的结果向量v
,用于存储最终的交集结果。 -
遍历第一个
使用范围set
:for
循环遍历a1
中的每个元素x
。 -
检查元素是否在第二个
对于set
中:a1
中的每个元素x
,使用a2.count(x)
检查该元素是否存在于a2
中。如果存在(即count
返回值大于0),则将该元素添加到结果向量v
中。 -
返回结果:
遍历完成后,返回包含交集元素的结果向量v
。
哈希:
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2)
{
unordered_set<int> s(nums1.begin(), nums1.end());
vector<int> result;
for (int num : nums2)
{
if (s.count(num))
{
result.push_back(num);
s.erase(num); // 确保每个元素只出现一次
}
}
return result;
}
};
首先将第一个数组nums1
的所有元素存入一个哈希表s
中。然后遍历第二个数组nums2
,如果nums2
中的元素在哈希表s
中存在,就将其添加到结果数组result
中,并从哈希表中删除该元素,以确保每个元素只出现一次。这种方法的时间复杂度为O(n + m),其中n和m分别是两个数组的长度。