题目描述
Given two arrays, write a function to compute their intersection.
Example:
Given nums1 = [1, 2, 2, 1]
, nums2 = [2,
2]
, return [2]
.
Note:
- Each element in the result must be unique.
- The result can be in any order.
解题思路
题目要求求两个数组的交集,并且交集中的元素不重复。
这道题是在sort分类中选择的,因此考虑用排序的方法来实现。
将两个数组都先排成有序数列,从小到大,用两个指针来表示当前所指到元素在数列中的位置。如果指的两个数大小不相同,则根据大小向后移动其中的一个指针,移动数字较小的那一个;如果两个数字相等,并且这个数字与结果数组中的最后一个数不想等时,则说明没有重复,则将这个数字加入到结果数组中。需要注意的是,结果数组刚开始为空,因此第一次的判断应该是当数组为空时,就将相等的这个数加入,不需要满足“这个数字与结果数组中的最后一个数不想等”这个条件。
复杂度分析
排序所需要的复杂度为O(N*logN),移动指针比较需要时间复杂度为O(N),总复杂度也为O(N*logN)
代码如下:
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
sort(nums1.begin(), nums1.end());
sort(nums2.begin(), nums2.end());
vector<int>res;
int i=0,j=0;
while (i<nums1.size()&&j<nums2.size()) {
if(nums1[i]>nums2[j]){
j++;
}
else if (nums1[i]<nums2[j]){
i++;
}
else{
if (res.size()==0||nums1[i]!=res.back()){
res.push_back(nums1[i]);
}
i++;
j++;
}
}
return res;
}
};