【每日刷题】两个数组的交集

day10, 两个数组的交集

题目来源:leetcode
给定两个数组,编写一个函数来计算它们的交集。

示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: 2

示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]
说明:

输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。

解答:刚开始想用一个hash表来记录重复元素,可是仔细一想行不通,因为如果有负数存在处理很麻烦,并且一旦数组开的很大,会有溢出问题。
之后便用set解决问题,开辟两个set容器,一个用来存nums1的元素并去重,另一个解决交集问题。

代码:

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        
        set<int> number, number_t;
        vector<int> num;
        
        for( int i = 0; i < nums1.size(); i++)
            number.insert( nums1[i]);
        for( int i = 0; i < nums2.size(); i++)
            if( number.count( nums2[i]) > 0)
                number_t.insert( nums2[i]);
    
        for( set<int>::iterator it = number_t.begin(); it != number_t.end(); it++)
               num.push_back( *it);
                
        return num;
    }
};

运行结果:【每日刷题】两个数组的交集_第1张图片


我的微信公众号

你可能感兴趣的:(每日刷题)