Leetcode-349. 两个数组的交集

给定两个数组，写一个函数来计算它们的交集。

例子:

 给定 num1= [1, 2, 2, 1], nums2 = [2, 2], 返回 [2].

提示:

• 每个在结果中的元素必定是唯一的。
• 我们可以不考虑输出结果的顺序。

解答： 

先将两个数组从小到大排序，然后以短的数组为基准进行遍历，使用一个指针ptr指向>=short_array[i]的第一个数，若两者相等，则存入结果中（当然，得和结果数组已有元素不相同），否则不加入结果数组中。

时间复杂度 排序：O(nlog(n))  查找：O(n)  n为长数组长度。

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        int ptr1 = 0;
        int ptr2 = 0;
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        int flag;
        int cnt = 0;
        flag = nums1.size() >= nums2.size() ? 1 : 2;
        vector<int> ret;
        if (flag == 1) {
            for (int i = 0; i<nums2.size(); i++) {
                while (ptr1<nums1.size()-1&&nums1[ptr1]<nums2[i]) ptr1++;
                if (nums1[ptr1] == nums2[i]) {
                    if (cnt == 0 || (cnt>0&&ret[cnt - 1] != nums2[i])) {
                        ret.push_back(nums2[i]);
                        cnt++;
                    }
                }
            }
        }
        if (flag == 2) {
            for (int i = 0; i<nums1.size(); i++) {
                while (ptr2<nums2.size()-1&&nums2[ptr2]<nums1[i]) ptr2++;
                if (nums2[ptr2] == nums1[i]) {
                    if (cnt == 0 || (cnt>0 && ret[cnt - 1] != nums1[i])) {
                        ret.push_back(nums1[i]);
                        cnt++;
                    }
                }
            }
        }
        return ret;
    }
};