本题使用了哈希表(unordered set, 数组)两种数据结构,关键点在于掌握unordered set的特性和set容器的成员方法。
时间复杂度:O(n),空间复杂度:O(n)
思路:
- 创建一个unordered_set来存储结果,再创建一个unordered_set来存放第一个数组,初始化时就将数组传入。注意:unordered set会自动给结果降重。
- 遍历第二个数组,如果找到了nums2中的元素在 nums1 中出现,就将它添加到结果集合中。注意:find函数如果找到了就返回迭代器,否则就返回num_set.end(),因此此处的不等于就相当于找到了。
- 最后将结果集合转换为 vector 并返回。
无序集合代码:
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_set<int> result_set; // 存放结果,并且set可以给结果集去重
unordered_set<int> nums_set(nums1.begin(), nums1.end());
for (int i = 0; i < nums2.size(); i++) {
if (nums_set.find(nums2[i]) != nums_set.end()) { // 发现nums2的元素 在nums_set里出现了
result_set.insert(nums2[i]);
}
}
return vector<int>(result_set.begin(), result_set.end());
}
};
时间复杂度:O(n),空间复杂度:O(1)
思路:
- 创建一个unordered_set来存储结果,再创建一个长度为1005的数组nums来存储数组nums1中出现的元素。注意:unordered set会自动给结果降重。
- 遍历第一个数组,如果某个元素出现过,就将nums数组相应的下标对应的元素设为1。
- 遍历第二个数组,对于其中的每个元素,如果在nums数组中对应的下标的元素为 1,就将它添加到结果集合中。
- 最后将结果集合转换为 vector 并返回。
数组代码:
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_set<int> result_set; // 存放结果,并且set可以给结果集去重
int nums[1005] = {0};
for (int i = 0; i < nums1.size(); i++) {
nums[nums1[i]] = 1;
}
for (int i = 0; i < nums2.size(); i++) {
if (nums[nums2[i]] == 1) {
result_set.insert(nums2[i]);
}
}
return vector<int>(result_set.begin(), result_set.end());
}
};