C++ 实现求两个数组的交集

最近发现了点好东西，小浩算法系列的文章都是精品，值得阅读学习，上一篇记录的map容器，其实就是在阅读了实现两个数组的交集的文章之后去填补的知识。

一、简单实现

这是原文的结题思路，也是为什么需要补充map知识点的原因，需要应用到map容器进行处理：

首先拿到这道题，我们基本马上可以想到，此题可以看成是一道传统的映射题（map映射），为什么可以这样看呢，因为我们需找出两个数组的交集元素，同时应与两个数组中出现的次数一致。这样就导致了我们需要知道每个值出现的次数，所以映射关系就成了<元素,出现次数>。剩下的就是顺利成章的解题。

文章中提供了go语言的解法，还说很简单，我......可能是太菜了，看了半天没理解，迷迷糊糊照搬实现了C++版本的代码。关于map容器，在使用的时候，如果重复添加了相同键名的元素，是会进行覆盖的，所以要使用multimap容器，它们的使用是差不多的。

#include <iostream>
#include <string>
#include <map>

using namespace std;

int main() {

    int num1[] = {1,2,2,1};
    int num2[] = {2,3,4,2,1};
    multimap<int, int> num_map;

    //范围for循环遍历数组1
    for (int &i : num1) {
        pair<int, int> value(i, 1);
        num_map.insert(value);
    }

    //遍历数组2
    int k = 0;
    for (int &v :num2) {
        multimap<int, int>::iterator iterator_num2 = num_map.find(v);
        if (iterator_num2 != num_map.end()) {
            iterator_num2->second -= 1;
            num2[k] = v;
            k++;
        }
    }
    
    for (int j = 0; j < k; j++) {
        cout << num2[j] << endl;
    }

    return 0;
}

在这里我发现了一个问题，当num2含有的交集元素比num1含有的多的时候，这种算法就会出错，例如修改成下面这样：

    int num1[] = {1,2,1,2};
    int num2[] = {2,3,2,2,1};

输出的结果为：

2
2
2
1

这个结果显然是不对的，出现这个结果的原因，其实是find返回的值是第一个与键名对应的实例，我阅读了博文C++ multimap查找相同键的键值对方法中提供的查找重复键值对的方法，但是并没有解决实际的问题，因为一开始就出错了，使用multimap并没有实现<元素，出现次数>的映射。

二、修正

事实上使用map就可以了，而不需要使用multimap，利用map处理好元素与出现次数的关系后，后面就容易处理了，后来我到LeetCode上面测试了一下我的代码，发现存在许多问题，测试用例是很重要的，想起一句话，出现错误是不可怕的，可怕的是不知道为什么对了，然后就坚持自己原本错误的答案为正确。下面这里是我通过测试的代码：

#include <iostream>
#include <string>
#include <map>
#include <vector>

using namespace std;

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> result;
        map<int, int> num_map;

        //范围for循环遍历数组1
        for (int& i : nums2) {
            map<int, int>::iterator  iter = num_map.find(i);
            if (iter != num_map.end()) {
                size_t cnt = iter->second;
                iter->second = ++cnt;
            }
            else {
                pair<int, int> value(i, 1);
                num_map.insert(value);
            }
        }

        //遍历数组2
        for (int v : nums1) {
            map<int, int>::iterator iterator_num2 = num_map.find(v);
            if (iterator_num2 != num_map.end() && (iterator_num2->second > 0)) {
                iterator_num2->second -= 1;
                result.push_back(v);
            }
        }
        return result;
    }
};


int main() {

    vector<int> num1{21, 47, 80, 4,  3,  24, 87, 12, 22, 11,\
                     48, 6,  89, 80, 74, 71, 74, 55, 58, 56,\
                     4,  98, 40, 66, 49, 53, 62, 27, 3,  66,\
                     78, 24, 48, 69, 88, 12, 80, 63, 98, 65,\
                     46, 35, 72, 5,  64, 72, 7,  29, 37, 3, \
                     2,  75, 44, 93, 79, 78, 13, 39, 85, 26,\
                     15, 41, 87, 47, 29, 93, 41, 74, 6,  48,\
                     17, 89, 27, 61, 2,  68, 99, 57, 45, 73,\
                     25, 33, 38, 32, 58};
    vector<int> num2{1,  39, 6,  81, 85, 10, 38, 22, 0,  89,\
                     57, 93, 58, 69, 65, 80, 84, 24, 27, 54,\
                     37, 36, 26, 88, 2,  7, 24,  36, 51, 5, \
                     74, 57, 45, 56, 55, 67, 25, 33, 78, 49,\
                     16, 79, 74, 80, 36, 27, 89, 49, 64, 73,\
                     96, 60, 92, 31, 98, 72, 22, 31, 0,  93,\
                     70, 87, 80, 66, 75, 69, 81, 52, 94, 90,\
                     51, 90, 74, 36, 58, 38, 50, 9,  64, 55,\
                     4,  21, 49, 60, 65, 47, 67, 8,  38, 83 };

    Solution test;
    vector<int> num = test.intersect(num1, num2);
    for (vector<int>::iterator iter = num.begin(); iter != num.end(); iter++) {
        cout << *iter << " ";
    }

    return 0;
}

三、小结

这个代码还能够再优化，而且题目还提出了一个进阶的问题，因为花了不少时间，所以暂时就做到这里了，有空再看看、、、