1. 程式人生 > >【LeetCode】350. Intersection of Two Arrays II(兩個陣列的交集 II)-map實現

【LeetCode】350. Intersection of Two Arrays II(兩個陣列的交集 II)-map實現

問題描述:

實現程式碼:.

#include <iostream>
#include <vector>
#include <map>

using namespace std;

// 時間複雜度: O(nlogn)
// 空間複雜度: O(n)
class Solution {
public:
	vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {

		map<int, int> record;//第一個引數表示關鍵字,
							 //第二個引數是關鍵字對應的值
		                     //record作為計數
		for (int i = 0; i < nums1.size(); i++)
			record[nums1[i]] += 1; //提取nums1[i]的計數器並將其加1

		vector<int> resultVector;
		for (int i = 0; i < nums2.size(); i++)
			if (record[nums2[i]] > 0) {  //如果計數record[nums2[i]]>0
				resultVector.push_back(nums2[i]);
				record[nums2[i]] --;  //nums2[i]的計數減1
			}

 		return resultVector;
	}
};

int main() {

	int nums1[] = { 1, 2, 2, 1 };
	vector<int> vec1(nums1, nums1 + sizeof(nums1) / sizeof(int));

	int nums2[] = { 2, 2 };
	vector<int> vec2(nums2, nums2 + sizeof(nums2) / sizeof(int));

	vector<int> res = Solution().intersect(vec1, vec2);
	for (int i = 0; i < res.size(); i++)
		cout << res[i] << " ";
	cout << endl;

 	return 0;
}

圖解:

①將陣列num1的元素新增到關聯容器map的 record中:

②查詢交集

 關於map: