This commit is contained in:
youngyangyang04
2020-07-02 09:42:03 +08:00
parent 7201721352
commit 1a3fe9539f

View File

@ -11,14 +11,15 @@ https://leetcode-cn.com/problems/intersection-of-two-arrays/
class Solution { class Solution {
public: public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) { vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_set<int> set; // 存放结果 unordered_set<int> result_set; // 存放结果
unordered_set<int> nums_set(nums1.begin(), nums1.end()); unordered_set<int> nums_set(nums1.begin(), nums1.end());
for (int num : nums2) { for (int num : nums2) {
if (nums_set.find(num) != nums_set.end()) { // 发现nums1与nums2有重复 // 发现nums2的元素 在nums_set里又出现过
set.insert(num); if (nums_set.find(num) != nums_set.end()) {
result_set.insert(num);
} }
} }
return vector<int>(set.begin(), set.end()); return vector<int>(result_set.begin(), result_set.end());
} }
}; };
``` ```