问题描述:
返回两个数组的交
注意事项
- 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> lz;
vector<int>::iterator it1;
vector<int>::iterator it2;
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());
int a=nums1.size();
int b=nums2.size();
int i=0;int j=0;
while(i<a&&j<b){
if(nums1[i]==nums2[j]){
lz.push_back(nums1[i]);
i++; j++;
}
if(nums1[i]<nums2[j]) i++;
if(nums1[i]>nums2[j]) j++;
}
return lz;
}
};
解题感悟:
排序后删除重复元素时用unique函数,用for循环判重时出现Runtime Error。