两数组的交

来源:互联网 发布:淘宝贷款买家改卖家 编辑:程序博客网 时间:2024/05/03 09:06

两数组的交 

返回两个数组的交

 注意事项
  • 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;
    }
};