[LeetCode] 349. Intersection of Two Arrays(两个数组的交集)-使用set

本题是TWO SIGMA公司的面试题。

题目描述：

使用关联容器set求解。 

 实现代码如下：

#include <iostream>
#include <vector>
#include <set>

using namespace std;

// 时间复杂度: O(nlogn)
// 空间复杂度: O(n)
class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {

        set<int> record;  //定义一个关联容器set,set中不包含重复的元素
        for( int i = 0 ; i < nums1.size() ; i ++ )
            record.insert(nums1[i]);  //将num1中的元素放入到record中
		                              //若一个元素已经存在，则insert操作没有影响

        set<int> resultSet;  //定义一个关联容器存放结果值
        for( int i = 0 ; i < nums2.size() ; i ++ )
            if(record.find(nums2[i]) != record.end()) //在record查找num2[i]元素，
		                                 //如果找到了，则将num2[i]插入到resultSet
                resultSet.insert(nums2[i]);

        vector<int> resultVector;  //因为需要返回的结果是vector<int>,
		                           //所以要定义一个vector容器
        for(set<int>::iterator iter = resultSet.begin(); iter != resultSet.end(); iter ++ )
            resultVector.push_back(*iter);   //将结果放到resultVector

        return resultVector;
    }
};

int main() {

    int nums1[] = {1, 2, 2, 1};
    vector<int> vec1(nums1, nums1 + sizeof(nums1)/sizeof(int));
		          //表示vec1有num1个元素，
	              //每个值是nums1 + sizeof(nums1) / sizeof(int)
	              //sizeof(nums1)为数组nums1所占的字节数，等于4*4=16
	              //sizeof(int)为一个元素所占的字节数，等于4

    int nums2[] = {2, 2};
    vector<int> vec2(nums2, nums2 + sizeof(nums2)/sizeof(int));

    vector<int> res = Solution().intersection(vec1, vec2);
    for(int i = 0 ; i < res.size() ; i ++ )
        cout << res[i] << " ";
    cout << endl;

    return 0;
}

图解：

①

②

将代码简化：

class Solution {
public:
	vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {

		set<int> record(nums1.begin(), nums1.end());//nums1.begin()表示第一个元素；
		                                //nums1.end()表示最后一个元素的下一个位置

		set<int> resultSet;
		for (int i = 0; i < nums2.size(); i++)
			if (record.find(nums2[i]) != record.end())
				resultSet.insert(nums2[i]);

		return vector<int>(resultSet.begin(), resultSet.end());
	}
};

< 本文完>