两数组的交
返回两个数组的交
注意事项
- Each element in the result must be unique.
- The result can be in any order.
nums1 = [1, 2, 2, 1]
, nums2 = [2, 2]
, 返回 [2]
.
解题思路:用两个map记录一下每个数对应的个数,然后用nums1进行遍历一遍,若果对应nums2的map里的数不为0则加入向量,最终返回向量。
解题代码:
class Solution {
public:
/**
* @param nums1 an integer array
* @param nums2 an integer array
* @return an integer array
*/
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
// Write your code here
map<int,int>s1,s2;
for(int i=0;i<nums1.size();i++)s1[nums1[i]]++;
for(int i=0;i<nums2.size();i++)s2[nums2[i]]++;
vector<int>ans;
for(int i=0;i<nums1.size();i++){
if(s2[nums1[i]])ans.push_back(nums1[i]),s2[nums1[i]]=0;
}
return ans;
}
};