哈希表-数组交集

楚楚楚歌

已于 2022-09-08 21:42:41 修改

阅读量273

点赞数

分类专栏：代码随想录文章标签：散列表 leetcode 数据结构

于 2022-09-08 17:22:46 首次发布

本文链接：https://blog.csdn.net/qq_38422570/article/details/126768527

版权

代码随想录专栏收录该内容

32 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何使用不同的数据结构和算法来解决LeetCode上的两道题目——349.两个数组的交集和350.两个数组的交集II。作者首先展示了使用计数向量的方法，然后优化为使用哈希映射，以更高效地找到两个整数数组的交集。这种方法减少了时间复杂度，提高了代码效率。

摘要由CSDN通过智能技术生成

题目链接：349. 两个数组的交集 - 力扣（LeetCode）

题目链接：350. 两个数组的交集 II - 力扣（LeetCode）

题目链接：349. 两个数组的交集 - 力扣（LeetCode）

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> record(1001);
        vector<int> res;
        //本来想用++，但不行
        //看了题解才明白直接赋值得了
        for(int i=0;i<nums1.size();i++)
        {
            record[nums1[i]]=1;
        }
        for(int i=0;i<nums2.size();i++)
        {
            if(record[nums2[i]]==1)
            {
                record[nums2[i]]=2;
            }
        }
        for(int i=0;i<record.size();i++)
        {
            if(record[i]==2)
            {
                res.push_back(i);
            }
        }
        return res;
    }
};

题目链接：350. 两个数组的交集 II - 力扣（LeetCode）

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        vector<int> record1(1001);
        vector<int> record2(1001);
        for(int i=0;i<nums1.size();i++)
        {
            record1[nums1[i]]++;
        }
        for(int i=0;i<nums2.size();i++)
        {
            record2[nums2[i]]++;
        }
        int j;
        for(int i=0;i<record1.size();i++)
        {
            j=1;
            while(record1[i]>0&&record2[i]>0&&j<=record2[i]&&j<=record1[i])
            {
                // cout<<record1[i]<<endl;
                res.push_back(i);
                j++;
            }
        }
        return res;
    }
};

看题解后

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        unordered_map<int,int> m;
        //计算nums1中的元素出现次数
        for(int i=0;i<nums1.size();i++)
        {
            m[nums1[i]]++;
        }
        //当nums2中的元素包含在nums1中时，把num加入结果数组，当nums1中的该元素出现次数-完之后，不再加入结果数组
        //因为nums2是挨个遍历的，所以nums2的元素次数<nums1时自动不会再加入结果数组了
        for(int num:nums2)
        {
            if(m.count(num))
            {
                res.push_back(num);
                m[num]--;
                if(m[num]==0)
                {
                    m.erase(num);
                }
            }
        }
        return res;
    }
};