1.题目
返回两个数组的交
注意事项
- Each element in the result must be unique.
- The result can be in any order.
nums1 = [1, 2, 2, 1]
, nums2 = [2, 2]
, 返回 [2]
.
2.思路
先对两个数组进行排序,然后从两个数组的头开始遍历并比对,当比对到相同元素时加入到交数组中。
因为是已经排过序的数组,所以当1数组的i位小于2数组的j位时,可以直接下移比对1数组的i+1位和2数组的j位;2数组j位小于1数组的i位时则2数组下移,比对2数组j+1位和1数组的i位。
这个问题的难点在于如果相同的数字出现两边则只保存一遍的。所以在把相同元素加入到交数组时,要进行判断是否已加入过该元素,若已经加入过,则不再加入第二遍。
3.代码
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> re;
int i=0;
int j=0;
int pre=-1;
sort(nums1.begin(),nums1.end());
sort(nums2.begin(),nums2.end());
while(i<nums1.size()&&j<nums2.size())
{
if(nums1[i]<nums2[j])
i++;
if(nums1[i]>nums2[j])
j++;
if(nums1[i]==nums2[j]&&nums1[i]!=pre)
{
re.push_back(nums1[i]);
pre=nums1[i];
i++; j++;
}
if(nums1[i]==nums2[j]&&nums1[i]==pre)
{ i++; j++;}
}
return re;
}
};
4.感想
这个问题的判断是否已经加入过该元素很有意思。一开始的想法是不管三七二十一先将相同元素加入到交数组中,再遍历交数组删除重复元素,但是觉得那样好像复杂了。。不如直接在加入元素时进行判断:在每次加入时用临时变量保存其值,下次加入时与临时变量进行比较,如果相同则不再重复加入,这样就简单多了,省的再费第二遍事。。。