附原题链接:https://leetcode-cn.com/problems/intersection-of-two-arrays
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]
示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
说明:
- 输出结果中的每个元素一定是唯一的。
- 我们可以不考虑输出结果的顺序。
此题直接用暴力破解法,三个for()循环判断可解,但运行时间32ms,击败了个位数的用户。
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
vector<int> result;
int k;
for (int i=0;i<nums1.size();i++){
for (int j=0;j<nums2.size();j++){
if (nums1[i] == nums2[j]){
for (k=0;k<result.size();k++){
if (nums1[i] == result[k]){
break;
}
}
if (k == result.size()){
result.push_back(nums1[i]);
}
}
}
}
return result;
}
};后来看了大佬的做法,用哈希数据结构:unordered_set(set中每一个元素都是唯一的,没有重复。)运行时间8ms。
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_set<int> result_set; // 存放结果
unordered_set<int> nums_set(nums1.begin(), nums1.end()); //用nums1初始化set
for (int num : nums2) {
if (nums_set.find(num) != nums_set.end()) { // 发现nums2的元素 在nums_set里又出现过
result_set.insert(num);
}
}
return vector<int>(result_set.begin(), result_set.end());
}
};
版权声明:本文为weixin_43908642原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。