试解leetcode算法题--两个数组的交集

<题目表述>

给定两个数组，编写一个函数来计算它们的交集。

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]
示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]

<原题链接>

https://leetcode-cn.com/problems/intersection-of-two-arrays

<思路>

初步思路是利用STL - set集合的特性，先将一个数组存放到集合中（此时相同的元素已被去除），再次遍历第二个数组，并在集合中查找，如果找到了该元素，那么将其压入到存放结果的向量中去，后将结果向量返回即可。

<样例代码>

#include<iostream>
#include<vector>
#include<algorithm>
#include<set>
using namespace std;
class Solution {
public:
	vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
		set<int> s;
		set<int>::iterator si;
		vector<int> ans;
		
        for (vector<int>::iterator vi = nums1.begin(); vi != nums1.end(); vi++)
			s.insert(*vi);
        
		sort(nums2.begin(), nums2.end());
		for (vector<int>::iterator vi = nums2.begin(); vi != nums2.end(); vi++)
		{
			if (s.find(*vi) != s.end() && (ans.empty() || *vi != ans[ans.size()-1]))
				ans.push_back(*vi);
		}
		return ans;
	}
};