关闭

[leetcode] 350. Intersection of Two Arrays II

标签: leetcodeeasy
62人阅读 评论(0) 收藏 举报
分类:

Given two arrays, write a function to compute their intersection.

Example:
Given nums1 = [1, 2, 2, 1]nums2 = [2, 2], return [2, 2].

Note:

  • Each element in the result should appear as many times as it shows in both arrays.
  • The result can be in any order.

Solution

Idea: hash table O(m + n) Space: O(m + n)
class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ret;
        unordered_map<int,int> h;
        for(int i=0; i<nums1.size();i++)
            ++h[nums1[i]];
        
        for(int j=0; j<nums2.size();j++){
            if(h[nums2[j]]>0){
                ret.push_back(nums2[j]);
                h[nums2[j]]--;
            }
        }
        return ret;
        
    }
};

Follow up 1: What if the given array is already sorted? How would you optimize your algorithm? 
Solution: O(max(m, n) log(max(m, n))) Space: O(m + n)
class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ret;
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        int i = 0,j = 0;
        while (i<nums1.size() && j<nums2.size()){
            if (nums1[i]==nums2[j]){
                ret.push_back(nums1[i]);
                i++; 
                j++;
            }
            else if (nums1[i]>nums2[j]) j++;
            else i++;
        }
        return ret;
    }
};



0
0

猜你在找
【直播】机器学习&数据挖掘7周实训--韦玮
【套餐】系统集成项目管理工程师顺利通关--徐朋
【直播】3小时掌握Docker最佳实战-徐西宁
【套餐】机器学习系列套餐(算法+实战)--唐宇迪
【直播】计算机视觉原理及实战--屈教授
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之矩阵--黄博士
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之凸优化--马博士
【套餐】Javascript 设计模式实战--曾亮
查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:9654次
    • 积分:632
    • 等级:
    • 排名:千里之外
    • 原创:31篇
    • 转载:150篇
    • 译文:0篇
    • 评论:0条
    文章分类