LeetCode 349 两个数组的集合(双指针 + set)

题目

题目链接
给定两个数组，编写一个函数来计算它们的交集。

示例 1：

输入：nums1 = [1,2,2,1], nums2 = [2,2]
输出：[2]
示例 2：

输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[9,4]

说明：

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

解题思路一

对两个数组排序后，通过双指针不断比较，相等时只有ans为空或者ans中最后一个元素不等于当前元素时才放入ans（防止重复元素放入）

代码

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        sort(nums1.begin(),nums1.end());
        sort(nums2.begin(),nums2.end());
        vector<int> ans;
        int i = 0,j = 0;
        while(i < nums1.size() && j < nums2.size()){
            if(nums1[i] < nums2[j])
                i++;
            else if(nums1[i] == nums2[j]){
                if(ans.empty() || nums1[i] != ans.back()){//此处判断是否会放入重复元素
                    ans.push_back(nums1[i]);
                }
                i++;j++;
            }else   
                j++;
        }
        return ans;
    }
};

解题思路二

使用set达到去重的功能，用find函数完成查找交集的功能

代码

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ans;
        set<int> st,st1;
        for(int i = 0;i < nums1.size();i++)
            st.insert(nums1[i]);
        for(int i = 0;i < nums2.size();i++)
            st1.insert(nums2[i]);
        for(set<int>::iterator it = st.begin();it != st.end();it++){
            if(st1.find(*it) != st1.end())
                ans.push_back(*it);
        }
        return ans;
    }
};

这里不需要排序所以用unordered_set完成去重功能，降低复杂度，同时用初始化的技巧减少代码量。

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> ans;
        unordered_set<int> st(nums2.begin(),nums2.end());
        for(int i = 0;i < nums1.size();i++){
            if(st.find(nums1[i]) != st.end()){
                ans.insert(nums1[i]);
            }
        }
        return vector<int>(ans.begin(),ans.end());
    }
};