力扣-349. 兩個陣列的交集

DawnTraveler發表於2024-05-03
力扣陣列

1.題目

題目地址(349. 兩個陣列的交集 - 力扣（LeetCode）)

https://leetcode.cn/problems/intersection-of-two-arrays/

題目描述

給定兩個陣列 nums1 和 nums2 ，返回 它們的

。輸出結果中的每個元素一定是 唯一 的。我們可以 不考慮輸出結果的順序 。

示例 1：

輸入：nums1 = [1,2,2,1], nums2 = [2,2]
輸出：[2]

示例 2：

輸入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
輸出：[9,4]
解釋：[4,9] 也是可透過的

提示：

  • 1 <= nums1.length, nums2.length <= 1000
  • 0 <= nums1[i], nums2[i] <= 1000

2.題解

2.1 雜湊集合

思路

使用unordered_set進行去重,然後檢視交集即可

程式碼

  • 語言支援：C++

C++ Code:


class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> st1, st2;
        vector<int> ans;
        for(int &num : nums1){
            st1.emplace(num);
        }
        for(int &num : nums2){
            st2.emplace(num);
        }

        for(const int &num : st1){
            if(st2.count(num)){
                ans.emplace_back(num);
            }
        }
        return ans;
    }
};

複雜度分析

  • 時間複雜度：\(O(m+n)\),其中\(m\)和\(n\)分別是兩個陣列的長度。
    使用兩個集合分別儲存兩個陣列中的元素需要\(O(m+n)\)的時間，
    遍歷較小的集合並判斷元素是否在另一個集合中需要\(O(\min(m,n))\)的時間，
    因此總時間複雜度是\(O(m+n)\) 。
  • 空間複雜度：\(O(m+n)\),
    其中\(m\)和\(n\)分別是兩個陣列的長度。
    空間複雜度主要取決於兩個集合。

2.2 排序+雙指標

思路

程式碼

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        int length1 = nums1.size(), length2 = nums2.size();
        int index1 = 0, index2 = 0;
        vector<int> intersection;
        while (index1 < length1 && index2 < length2) {
            int num1 = nums1[index1], num2 = nums2[index2];
            if (num1 == num2) {
                // 保證加入元素的唯一性
                if (!intersection.size() || num1 != intersection.back()) {
                    intersection.push_back(num1);
                }
                index1++;
                index2++;
            } else if (num1 < num2) {
                index1++;
            } else {
                index2++;
            }
        }
        return intersection;
    }
};

複雜度分析

  • 時間複雜度：\(O(m\log m+n\log n)\),其中\(m\)和\(n\)分別是兩個陣列的長度。
    對兩個陣列排序的時間複雜度分別是\(O(m\log m)\)和\(O(n\log n)\) ,
    雙指標尋找交集元素的時間複雜度是\(O(m+n)\) ,
    因此總時間複雜度是\(O(m\log m+n\log n)\)。
  • 空間複雜度：\(O(\log m+\log n)\),其中\(m\)和\(n\)分別是兩個陣列的長度。
    空間複雜度主要取決於排序使用的額外空間。

相關文章