[LeetCode] Intersection of Two Arrays 兩個數組相交

 

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.

 

這道題讓我們找兩個數組交集的部分（不包含重復數字），難度不算大，我們可以用個set把nums1都放進去，然后遍歷nums2的元素，如果在set中存在，說明是交集的部分，加入結果的set中，最后再把結果轉為vector的形式即可：

 

解法一：

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        set<int> s(nums1.begin(), nums1.end()), res;
        for (auto a : nums2) {
            if (s.count(a)) res.insert(a);
        }
        return vector<int>(res.begin(), res.end());
    }
};

 

我們還可以使用兩個指針來做，先給兩個數組排序，然后用兩個指針分別指向兩個數組的開頭，然后比較兩個數組的大小，把小的數字的指針向后移，如果兩個指針指的數字相等，那么看結果res是否為空，如果為空或者是最后一個數字和當前數字不等的話，將該數字加入結果res中，參見代碼如下：

 

解法二：

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        int i = 0, j = 0;
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        while (i < nums1.size() && j < nums2.size()) {
            if (nums1[i] < nums2[j]) ++i;
            else if (nums1[i] > nums2[j]) ++j;
            else {
                if (res.empty() || res.back() != nums1[i]) {
                    res.push_back(nums1[i]);
                }
                ++i; ++j;
            }
        }
        return res;
    }
};

 

我們還可以使用二分查找法來做，思路是將一個數組排序，然后遍歷另一個數組，把遍歷到的每個數字在排序號的數組中用二分查找法搜索，如果能找到則放入結果set中，這里我們用到了set的去重復的特性，最后我們將set轉為vector即可：

 

解法三：

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        set<int> res;
        sort(nums2.begin(), nums2.end());
        for (auto a : nums1) {
            if (binarySearch(nums2, a)) {
                res.insert(a);
            }
        }
        return vector<int>(res.begin(), res.end());
    }
    bool binarySearch(vector<int> &nums, int target) {
        int left = 0, right = nums.size();
        while (left < right) {
            int mid = left + (right - left) / 2;
            if (nums[mid] == target) return true;
            else if (nums[mid] < target) left = mid + 1;
            else right = mid;
        }
        return false;
    }
};

 

或者我們也可以使用STL的set_intersection函數來找出共同元素，很方便：

 

解法四：

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        set<int> s1(nums1.begin(), nums1.end()), s2(nums2.begin(), nums2.end()), res;
        set_intersection(s1.begin(), s1.end(), s2.begin(), s2.end(), inserter(res, res.begin()));
        return vector<int>(res.begin(), res.end());
    }
};

 

類似題目：

Intersection of Two Arrays II

 

參考資料：

https://leetcode.com/discuss/103345/three-java-solutions

https://leetcode.com/discuss/103224/my-c-solution-with-sort

https://leetcode.com/discuss/103295/my-c-solutions-using-set-and-unordered_set

 

LeetCode All in One 題目講解匯總(持續更新中...)