文章目录
- 一、题目
- 二、解法
- 三、完整代码
所有的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 ) O(m+n) O(m+n), m和n分别分两数组的长度,程序当中分别遍历了两数组。
- 空间复杂度: O ( t ) O(t) O(t),其中 t = m i n ( m , n ) t = min(m, n) 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