本文主要是介绍LeetCode 349. Intersection of Two Array,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
349. Intersection of Two Array
一、问题描述
Given two arrays, write a function to compute their intersection.
Note:
- Each element in the result must be unique.
- The result can be in any order.
二、输入输出
Example:
Given nums1 = [1, 2, 2, 1]
, nums2 = [2, 2]
, return [2]
.
三、解题思路
- 使用stl的库可以轻易的搞定这道题。set是集合,存储的是不重复的且有序的。可以使用unique来去重,但是之前需要先排序依次,他去除的只是相邻的重复。
- 先遍历nums1把元素存储到set中,然后遍历nums2.如果set中有,说明两个数组中都有这个元素。最后对返回的vector去重就行了
class Solution {
public:vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {set<int> pNums;vector<int> ret;for (auto ite : nums1){pNums.insert(ite);}for (auto ite : nums2){if(pNums.find(ite) != pNums.end()) ret.push_back(ite);}sort(ret.begin(), ret.end());vector<int>::iterator endUnique = unique(ret.begin(), ret.end());ret.erase(endUnique, ret.end());return ret;}
};
这篇关于LeetCode 349. Intersection of Two Array的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!