LintCode-Intersection of Arrays

最新推荐文章于 2020-06-01 21:03:48 发布

原创最新推荐文章于 2020-06-01 21:03:48 发布 · 218 阅读

0 ·

CC 4.0 BY-SA版权

lintcode 同时被 2 个专栏收录

14 篇文章

订阅专栏

c++

13 篇文章

订阅专栏

本文介绍了一种使用C++ STL中的set来高效计算多个数组交集的方法。通过将第一个数组的元素插入到一个set中，然后遍历后续数组，只保留存在于当前set中的元素，最终得到的set即为所有数组的交集。

给出多个数组，求它们的交集。输出他们交集的大小。

 注意事项
输入的所有数组元素总数不超过500000。
题目数据每个数组里的元素没有重复。题目不写

数据规模，还以为是什么没见过的算法，结果用set查找就行。
不用stl自己写的话估计要做个二叉搜索树什么的。

class Solution {
public:
    /**
     * @param arrs: the arrays
     * @return: the number of the intersection of the arrays
     */
    int intersectionOfArrays(vector<vector<int>> &arrs) {
        // write your code here
        set<int> a;
        if(arrs.size()==0) return 0;
        for(auto& var:arrs[0])
        {
            a.insert(var);
        }
        for(auto& array:arrs)
        {
            set<int> b;
            for(auto& var:array)
            {
                if(a.find(var)!=a.end())
                {
                    b.insert(var);
                }
            }
            a=move(b);
        }
        return a.size();
    }
};