C++ 中三個有序陣列的交集
假設有三個整數陣列 arr1、arr2 和 arr3,它們按嚴格遞增的順序排列,我們需要返回一個只包含這三個陣列中出現的整數的有序陣列。因此,如果陣列是 [1,2,3,4,5]、[1,2,5,7,9] 和 [1,3,4,5,8],則輸出將是 [1,5]
要解決此問題,我們將按以下步驟操作:
- 定義一個名為 res 的陣列
- 建立三個對映 f1、f2 和 f3
- 對於從 0 到 arr1 長度的 i
- f1[arr1[i]] 增加 1
- 對於從 0 到 arr2 長度的 i
- f2[arr2[i]] 增加 1
- 對於從 0 到 arr3 長度的 i
- f3[arr3[i]] 增加 1
- 對於 i = 1 到 2000,
- 如果 f1[i]、f2[i] 和 f3[i] 為真,則
- 把 i 插入 res 陣列
- 如果 f1[i]、f2[i] 和 f3[i] 為真,則
- 返回 res
示例
讓我們看看以下實現,以便更好地理解:
#include <bits/stdc++.h> using namespace std; void print_vector(vector<auto> v){ cout << "["; for(int i = 0; i<v.size(); i++){ cout << v[i] << ", "; } cout << "]"<<endl; } class Solution { public: vector<int> arraysIntersection(vector<int>& arr1, vector<int>& arr2, vector<int>& arr3) { vector <int> ans; unordered_map <int,int> f1,f2,f3; for(int i =0;i<arr1.size();i++){ f1[arr1[i]]++; } for(int i =0;i<arr2.size();i++){ f2[arr2[i]]++; } for(int i =0;i<arr3.size();i++){ f3[arr3[i]]++; } for(int i =1;i<=2000;i++){ if(f1[i] && f2[i] && f3[i])ans.push_back(i); } return ans; } }; main(){ Solution ob; vector<int> v1 = {1,2,3,4,5}; vector<int> v2 = {1,2,5,7,9}; vector<int> v3 = {1,3,4,5,8}; print_vector(ob.arraysIntersection(v1, v2, v3)); }
輸入
[1,2,3,4,5] [1,2,5,7,9] [1,3,4,5,8]
輸出
[1,5]
廣告