题目:两数组的交
问题描述:
返回两个数组的交
注意事项
Each element in the result must be unique.
The result can be in any order.
样例
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
vector<int> output;
sort(nums1.begin(), nums1.end());
sort(nums2.begin(), nums2.end());
int n1 = nums1.size();
int n2 = nums2.size();
int i=0, j=0;
int last = 0;
while(i<n1 && j<n2)
{
if(nums1[i] < nums2[j])
i++;
else if(nums1[i] > nums2[j])
j++;
else
{
if( !output.size()||nums1[i]>last)
{
output.push_back(nums1[i]);
last = nums1[i];
}
i++;
j++;
}
}
return output;
}
};
感想 :
注意查重。