【LeetCode】350. Intersection of Two Arrays II(兩個陣列的交集 II)-map實現
阿新 • • 發佈:2018-12-13
問題描述:
實現程式碼:.
#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: