力扣 346 两个数组的交集 c++

class Solution {
public:
    vector intersection(vector& nums1, vector& nums2) {
            int a,b,c;
    vector result;
	sort(nums1.begin(),nums1.end());
    sort(nums2.begin(),nums2.end());
    nums1.erase(unique(nums1.begin(), nums1.end()), nums1.end());
    nums2.erase(unique(nums2.begin(), nums2.end()), nums2.end());
    a = nums1.size();
    b = nums2.size();
	    for (int i=0;i

这是我自己打的代码

 

下面是编译器测试代码

#include
using namespace std;
int main()
{
    int a,b,c;
    vector nums1,nums2,result;
    	while(cin>>a)
	{
		if(a==-1)
			break;
		else
		{
			nums1.push_back(a);
		}
	}

	    	while(cin>>a)
	{
		if(a==-1)
			break;
		else
		{
			nums2.push_back(a);
		}
	}
	sort(nums1.begin(),nums1.end());
    sort(nums2.begin(),nums2.end());
    nums1.erase(unique(nums1.begin(), nums1.end()), nums1.end());
    nums2.erase(unique(nums2.begin(), nums2.end()), nums2.end());
    a = nums1.size();
    b = nums2.size();
	    for (int i=0;i

 

这是我在评论中看到的代码

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

	vector ret;
	set s;
	for (int i = 0; i < nums1.size(); i++) 
	{
		if (find(nums2.begin(), nums2.end(), nums1[i]) != nums2.end()) 
		{
			s.insert(nums1[i]);
		}
	}
	ret.assign(s.begin(),s.end());
	return ret;
}
};

这思路是一样的 但是这代码量差距也太大了.set是类似vector的一种数据结构,并且有去除重复的功能,assign是把数据放入到vector中。find函数在用于查找不同数组中同样的数据用的方法真是简单暴力啊。以后要记得。

 

你可能感兴趣的:(算法)