问题描述:返回两个数组的交。
样例
nums1 = [1, 2, 2, 1]
, nums2 = [2, 2]
, 返回 [2]
.
解题思路:在上个题的基础上加个删除重复元素的函数。
实验代码:
class Solution {
public:
/**
* @param nums1 an integer array
* @param nums2 an integer array
* @return an integer array
*/
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
// Write your code here
sort(nums1.begin(), nums1.end());
sort(nums2.begin(), nums2.end());
int i=0, j=0;
int num1 = nums1.size();
int num2 = nums2.size();
vector<int> v;
while(i<num1 && j<num2)
{if(nums1[i] == nums2[j])
{
v.push_back(nums1[i]);
i++;
j++;
}
else if(nums1[i] > nums2[j])
{ j++;}
else
{ i++;}
}
v.erase(unique(v.begin(),v.end()),v.end());
return v;
}
};
个人感想:将vector中重复元素删除。