Computer >> 컴퓨터 >  >> 프로그램 작성 >> C++

C++에서 정렬된 세 배열의 교차점


세 개의 정수 배열 arr1, arr2 및 arr3이 있고 엄격하게 오름차순으로 정렬되어 있다고 가정하면 이 세 가지 배열 모두에 나타난 정수로만 정렬된 배열을 반환해야 합니다. . 따라서 배열이 [1,2,3,4,5], [1,2,5,7,9] 및 [1,3,4,5,8]이면 출력은 [1,5]가 됩니다. ]

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

  • res라는 배열 정의
  • 3개의 지도 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]이면
      • res 배열에 i 삽입
  • 반환 결과

예시

더 나은 이해를 위해 다음 구현을 살펴보겠습니다. −

#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]