[leetcode]Intersection of Two Arrays

유의할점

sort하고 erase

풀이

nums1와 nums2를 unique하게 만들고 교집합을 찾는다. 투포인터를 이용 두개의 포인터가 가르키는 값이 같을 경우에 정답에 포함한다.

코드

C++

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        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 i = 0;
        int j = 0;
        vector<int> res;
        while(i<nums1.size()&&j<nums2.size()){
            if(nums1[i]==nums2[j]){
                res.push_back(nums1[i]);
                i++;
                j++;
            }else if(nums1[i] > nums2[j]){
                j++;
            }else{
                i++;
            }
        }
        return res;
    }
};