【题目描述】
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.
寻找两个数集的交集,重复的数字不计算在内
先对这两个数集进行排序,利用两个指针分别指向这两个数集,如果对应的两个数集的数是相同的,则说明是交集中的一个数,要注意的一点,为了避免重复,在每次循环的开始都先将指针指向重复的数里的最后一个
【代码】
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
int len1=nums1.size();
int len2=nums2.size();
vector<int> ans;
if(len1==0||len2==0) return ans;
int i=0,j=0;
sort(nums1.begin(),nums1.end());
sort(nums2.begin(),nums2.end());
while(i<len1&&j<len2){
while(i<len1-1 && nums1[i]==nums1[i+1]) i++;
while(j<len2-1 && nums2[j]==nums2[j+1]) j++;
if(nums1[i]==nums2[j]){
ans.push_back(nums1[i]);
i++;
j++;
}
else{
if(nums1[i]<nums2[j]) i++;
else j++;
}
}
return ans;
}
};