LeetCode之旅:350. 两个数组的交集 II

题意:

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

示例 1:

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

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明:

输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
进阶:

如果给定的数组已经排好序呢?你将如何优化你的算法?
如果 nums1 的大小比 nums2 小很多,哪种方法更优?
如果 nums2 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?

解法一:

先排序,再移动双迭代器。

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        // 升序排序
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        
        vector<int> resVec;
        
        // 双迭代器
        vector<int>::iterator itNums1 = nums1.begin();
        vector<int>::iterator itNums2 = nums2.begin();
        
        while(itNums1 != nums1.end() && itNums2 != nums2.end()){
            if(*itNums1 < *itNums2)
                ++itNums1;
            else if(*itNums1 > *itNums2)
                ++itNums2;
            else{
                resVec.push_back(*itNums1);
                ++itNums1;
                ++itNums2;
            }
        }
        
        return resVec;
    }
};

C++库函数sort、qsort:https://blog.csdn.net/zzzmmmkkk/article/details/4266888

解法二:

由于需要考虑元素出现的次数,所以采用map存储和查找。

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        map<int,int> nums1Map;
        for(int i = 0; i < nums1.size(); i++){
            nums1Map[nums1[i]]++;
        }

        vector<int> resVec;
        for(int i = 0; i < nums2.size(); i++){
            if(nums1Map[nums2[i]] > 0){
                resVec.push_back(nums2[i]);
                nums1Map[nums2[i]]--;
            }                
        }
        return resVec;
    }
};
class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        map<int, int> nums1Map;
        for(auto num1 : nums1){
            nums1Map[num1]++;
        }
        
        vector<int> resVec;
        for(auto num2 : nums2){
            if(nums1Map[num2] > 0){
                resVec.push_back(num2);
                nums1Map[num2]--;
            }
        }
        return resVec;
    }
};

解法三:

给数组做一组标志位数组.

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        
        vector<int> resVec;
        // 标志位数组,保证:输出结果中每个元素出现的次数,
        // 应与元素在两个数组中出现的次数一致。
        vector<bool> bl(nums2.size());
        
        for(int i = 0; i < nums1.size(); ++i){
            for(int j = 0; j < nums2.size(); ++j){
                if(nums1[i] == nums2[j] && bl[j] == false){
                    resVec.push_back(nums1[i]);
                    bl[j] = true;
                    break;
                }
            }
        }
        
        return resVec;
    }
};
class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        
        vector<int> resVec;
        // 标志位数组,保证:输出结果中每个元素出现的次数,
        // 应与元素在两个数组中出现的次数一致。
        vector<bool> bl(nums2.size());

        for(auto num1 : nums1){
            int j = 0;
            for(auto num2 : nums2){
                if(num1 == num2 && bl[j] == false){
                    resVec.push_back(num1);
                    bl[j] = true;
                    break;
                }
                ++j;
            }
        }
        
        return resVec;
    }
};

解法一最优,解法二次之,解法三最慢。

你可能感兴趣的:(LeetCode之旅)