【算法与数据结构】349、LeetCode两个数组的交集

所有的LeetCode题解索引，可以看这篇文章——【算法和数据结构】LeetCode题解。

一、题目

二、解法

  思路分析：使用了一个哈希数组记录nums1出现的数字，然后遍历nums2，找到hash数组为1的值就是交集的元素。
  程序如下：

class Solution {
public:
	vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
		unordered_set<int> result;
		int hashTable[1005] = { 0 };	// 哈希数组
		for(int num : nums1) hashTable[num] = 1;	// nums1记录出现的数字
		for (int num : nums2) {
			if (hashTable[num]) {
				result.insert(num);
			}
		}
		return vector<int>(result.begin(), result.end());
	}
};

复杂度分析：

• 时间复杂度： $O(m+n)$， m和n分别分两数组的长度，程序当中分别遍历了两数组。
• 空间复杂度：$O(t)$，其中$t=min(m,n)$， 两数组的最大交集可能是较短的数组。

三、完整代码

# include <iostream>
# include <unordered_set>
# include <vector>
# include <string>
using namespace std;

class Solution {
public:
	vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
		unordered_set<int> result;
		int hashTable[1005] = { 0 };	// 哈希数组
		for(int num : nums1) hashTable[num] = 1;	// nums1记录出现的数字
		for (int num : nums2) {
			if (hashTable[num]) {
				result.insert(num);
			}
		}
		return vector<int>(result.begin(), result.end());
	}
};

void VectorGenerator(int arr[], int len, vector<int>& nums1) {
	for (int i = 0; i < len; i++) {
		nums1.push_back(arr[i]);
	}
}

void my_print(vector <int>& v, string msg)
{
	cout << msg << endl;
	for (vector<int>::iterator it = v.begin(); it != v.end(); it++)
	{
		cout << *it << "  ";
	}
	cout << endl;
}

int main()
{
	int arr1[] = { 4,9,5 };
	int arr2[] = { 9,4,9,8,4 };
	int arr1_len = sizeof(arr1) / sizeof(int);
	int arr2_len = sizeof(arr2) / sizeof(int);
	vector<int> nums1, nums2;
	VectorGenerator(arr1, arr1_len, nums1);
	VectorGenerator(arr2, arr2_len, nums2);
	my_print(nums1, "nums1:");
	my_print(nums2, "nums2:");
	Solution s1;
  	vector<int> result = s1.intersection(nums1, nums2);
	my_print(result, "result:");
	system("pause");
	return 0;
}

end