5.3 两个数组的交集

题目

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

示例 1：

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

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

说明：

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

来源：力扣（LeetCode）
链接: link.

思路

1. 用 unordered_set 来存储数组元素
2. for ( int num : nums2 ) : num 代表 nums2 的每一个元素
   若在 nums_set 找到和 nums2中一样的元素,就插入到 result_set 中去
3. 返值

具体代码实现(C++)

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        //1
        unordered_set<int> result_set; //存放结果
        unordered_set<int> nums_set(nums1.begin(), nums1.end()); //把 vector 存入 set
        //2
        for (int num : nums2) {
            if (nums_set.find(num) != nums_set.end()) { //若 nums_set为空则等于 end()
                result_set.insert(num);  
            }
        }
        //3
        return vector<int>(result_set.begin(), result_set.end()); //把 set 存入 vector
    }
};

模型（知识点）

1.unordered_set
unordered_set 是一种底层实现基于hashtable无序集合

用法参见 unordered_set用法

2.for (int num : nums2) { }
C++ 11 标准中，for 循环添加了一种全新的语法格式，如下所示：

for (declaration : expression){
//循环体
}
其中，两个参数各自的含义如下：

declaration：表示此处要定义一个变量，该变量的类型为要遍历序列中存储元素的类型。需要注意的是，C++ 11 标准中，declaration参数处定义的变量类型可以用 auto 关键字表示，该关键字可以使编译器自行推导该变量的数据类型。

expression：表示要遍历的序列，常见的可以为事先定义好的普通数组或者容器，还可以是用 {} 大括号初始化的序列。

总结:

1.通式
for(要遍历的数据类型 遍历变量 ： 遍历对象)
注：遍历的数据类型要和遍历的对象元素类型一致

2.范围for循环不能用于循环体中有改变容器大小的操作。举例子：循环体内不能向vector容器添加元素。

详见 C++11 for循环（基于范围的循环）详解

3.vector / iterator

参见 C++ vector 容器浅析

C++ STL 教程 (容器,迭代器)

C++迭代器（STL迭代器）iterator详解