LeetCode之两个数组的交集

介绍：给定两个数组，编写一个函数来计算它们的交集。

示例 1:

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

示例2：

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]

说明：

 - 输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。
 - 我们可以不考虑输出结果的顺序。

主要分析：可以建立一个map<int,int>的数据结构，key就是nums1数组中的数字，value就是对应数字的个数，然后建立map，用该map的key去匹配nums2中的数据，如果key对应该数据，就是重负的元素，然后value-1，继续进行查找，把每次查找的结果放在一个vec中即可。

源代码

#include<iostream>
#include<string>
#include<vector>
#include<algorithm>
#include <sstream>
#include<map>

using namespace std;

vector<int> intersect(vector<int>& nums1, vector<int>& nums2)
{
	map<int, int> MapNums;
	for (int i = 0; i < nums1.size(); ++i)
	{
		++MapNums[nums1[i]];
	}

	vector<int> vec;

	for (int j = 0; j<nums2.size(); ++j)
	{
		if (MapNums[nums2[j]]>0)
		{
			vec.push_back(nums2[j]);
			--MapNums[nums2[j]];
		}
	}
	return vec;
}

int main()
{
	vector<int>  nums1 = { 1, 2, 2, 1 }, nums2 = { 2, 2 },res;
	res = intersect(nums1,nums2);
	for (int i = 0; i < res.size(); i++)
	{
		cout << res[i] << endl;
	}
	system("pause");
	return 0;
}

除了C++版本，还有一个python版本：

def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """

        m = len(nums1)
        n = len(nums2)

        if m == 0 or n == 0:
            return []


        dicts = {}
        for i in nums1:
            if i in dicts:
                dicts[i] += 1
            else:
                dicts[i] = 1

        ret = []
        for j in nums2:
            if j in dicts and dicts[j] > 0:
                ret.append(j)
                dicts[j] -= 1

        return ret