题目来源:牛客
NC313 两个数组的交集
题目描述
给定两个整数数组分别为 nums1 , nums2 ,找到他们的公共元素并返回。
数据范围:1 ≤ nums1.length , nums2.length ≤ 1000
1 ≤ nums1[i] , nums2[i] ≤ 1000
示例1
输入: [1,2] , [2,2,2,2]
返回值:[2]
说明: 两个数组的公共元素只有 2
示例2
输入: [1,2,3] , [8,2,2,3,8]
返回值:[2,3]
说明: 两个数组的公共元素为 2 和 3 ,返回 [3,2] 也是一个正确答案
解析
这个题稍微有点歧义,公共元素还是很容易找到,但是,假如两个数组里都含有两个 2 ,那么公共元素到底有几个 2 呢,这题目并没有说清楚,但在这题里,多个 2 只算做一个。
这题思路很简单,遍历第一个数组,将第一个数组拥有的元素标记一下,在遍历第二个数组,若是第二数组标记的元素已经被第一个数组标记过,那么这个数就是公共元素。
如图,被两次标记的元素就是公共元素。用容器哈希表来标记肯定是小题大做了,逻辑简单,我们使用数组模拟即可。
代码实现
本体是接口型,只需要把答案返回即可。
class Solution { public: vector<int> intersection(vector<int>& nums1, vector<int>& nums2) { // 哈希数组,记录数字是否在nums1中出现 bool hash[1010] = { 0 }; // 将答案存到ret中 vector<int> ret; // 遍历nums1,将里面的元素存放到hash数组中 for (auto e : nums1) { // 出现过则标记为true hash[e] = true; } // 遍历nums2 for (auto e : nums2) { // 如果该值在hash数组中为真,说明该数字是公共数字 if (hash[e]) { // 将该数字存到ret中 ret.push_back(e); // 避免重复存入同一个数字,需要在hash数组中去除该值 hash[e] = false; } } // 返回答案 return ret; } };