day10, 两个数组的交集
题目来源:leetcode
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: 2
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]
说明:
输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。
解答:刚开始想用一个hash表来记录重复元素,可是仔细一想行不通,因为如果有负数存在处理很麻烦,并且一旦数组开的很大,会有溢出问题。
之后便用set解决问题,开辟两个set容器,一个用来存nums1的元素并去重,另一个解决交集问题。
代码:
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
set<int> number, number_t;
vector<int> num;
for( int i = 0; i < nums1.size(); i++)
number.insert( nums1[i]);
for( int i = 0; i < nums2.size(); i++)
if( number.count( nums2[i]) > 0)
number_t.insert( nums2[i]);
for( set<int>::iterator it = number_t.begin(); it != number_t.end(); it++)
num.push_back( *it);
return num;
}
};
运行结果: