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

C++에서 두 배열의 교차


배열이 ​​두 개 있다고 가정합니다. 그들의 교차점을 찾아야 합니다.

따라서 입력이 [1,5,3,6,9],[2,8,9,6,7]과 같으면 출력은 [9,6]

이 됩니다.

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

  • 두 개의 맵 mp1, mp2 정의

  • 배열 res

    정의
  • nums1의 x에 대해

    • (mp1[x] 1씩 증가)

  • x의 경우 nums2

    • (mp2[x] 1씩 증가)

  • mp1의 각 키-값 쌍 x에 대해

    • cnt :=0

    • cnt :=x 값과 mp2[x의 키]

      의 최소값
    • cnt> 0이면 -

      • res의 끝에 x의 키 삽입

  • 반환 해상도

예시

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

#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> intersection(vector<int>& nums1, vector<int>& nums2){
      unordered_map<int, int> mp1, mp2;
      vector<int> res;
      for (auto x : nums1)
         mp1[x]++;
      for (auto x : nums2)
         mp2[x]++;
      for (auto x : mp1) {
         int cnt = 0;
         cnt = min(x.second, mp2[x.first]);
         if (cnt > 0)
            res.push_back(x.first);
         }
         return res;
      }
};
main(){
   Solution ob;
   vector<int> v = {1,5,3,6,9}, v1 = {2,8,9,6,7};
   print_vector(ob.intersection(v, v1));
}

입력

{1,5,3,6,9},{2,8,9,6,7}

출력

[9, 6, ]