Leetcode 349. Intersection of Two Arrays

Given two arrays, write a function to compute their intersection.

Example:
Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2].

Note:

• Each element in the result must be unique.

• The result can be in any order.

思路：遍历nums1，如果找到nums2里没有的元素就删除，最后再遍历一次nums1，删除重复的元素。

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;

/***********************************************提交部分******************************************************/ 
class Solution
{
	public:
		vector<int> intersection(vector<int>& nums1, vector<int>& nums2)
		{
			for(vector<int>::iterator p=nums1.begin(); p!=nums1.end(); )
			{
				if(find(nums2.begin(),nums2.end(),*p)==nums2.end())//如果在nums1里找到nums2中没有的元素则删除 
				{
					vector<int>::iterator p1=p;
					p=nums1.erase(p1);
				}
				else
					p++;
			}
			for(vector<int>::iterator p=nums1.begin(); p!=nums1.end();)//删除nums1里重复的元素 
			{
				if(find(nums1.begin(),nums1.end(),*p)!=nums2.end()&&find(nums1.begin(),nums1.end(),*p)!=p)
				{
					vector<int>::iterator p1=p;
					p=nums1.erase(p1);
				}
				else
					p++;
			}
			return nums1;
		}
};
/***************************************************************************************************************/

int main()
{
	int n,m;
	vector<int> a,b,ans;
	while(scanf("%d%d",&n,&m)==2)
	{
		int tmp;
		a.clear();
		b.clear();
		ans.clear();
		for(int i=0; i<n; i++)
		{
			scanf("%d",&tmp);
			a.push_back(tmp);
		}
		for(int i=0; i<m; i++)
		{
			scanf("%d",&tmp);
			b.push_back(tmp);
		}
		Solution s;
		ans=s.intersection(a,b);
		for(int i=0; i<ans.size(); i++)
			printf("%d ",ans[i]);
		printf("\n");
	}
	return 0;
}