LC350 两个数组的交集II

Posted 2021-02-20 Lee先森的博客

先分别排序，搞两个指针，从头开始对两个数组进行扫描，如果一个小一个大，就让小的往后走。如果相等那就存到结果数组中，然后两个指针同时往后走，循环到有一个遍历完就可以结束。

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> temp, ans;
        int p1 = 0, p2 = 0;
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        while (p1 < nums1.size() && p2 < nums2.size()) {
            if (nums1[p1] < nums2[p2]) p1++;
            else if (nums1[p1] > nums2[p2]) p2++;
            else {
                ans.push_back(nums1[p1]);
                p1++;
                p2++;
            }
        }
        return ans;
    }
};