问题描述:返回两个数组的交
样例:nums1 = [1, 2, 2, 1], nums2 = [2, 2], 返回 [2].
解题思路:给了两个数组,求取两个数组的交集。我最先想到的是,先将这两个数组从小到大排序,然后逐个比较,有相同的就把它挑出来,放到一个新的vector 向量当中,直至比较完所有的元素。但是在比较的过程当中,要注意两个数组的长度可能不相等,这个时候应该分情况进行讨论。
实现代码:
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;
}
};
做题感想:在第一种情况与第二种情况的时候感觉很好理解,但是到了第三种情况的时候,刚开始的时候老写错,一直没有改变last 的值。后来才发现这个错误,所以在写程序的过程当中一定要仔细认真。