Computer >> Máy Tính >  >> Lập trình >> C ++

Giao điểm của hai mảng trong C ++


Giả sử chúng ta có hai mảng; chúng ta phải tìm các giao điểm của chúng.

Vì vậy, nếu đầu vào là [1,5,3,6,9], [2,8,9,6,7], thì đầu ra sẽ là [9,6]

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -

  • Xác định hai bản đồ mp1, mp2

  • Xác định res một mảng

  • cho x trong nums1

    • (tăng mp1 [x] lên 1)

  • cho x trong nums2

    • (tăng mp2 [x] lên 1)

  • cho mỗi cặp khóa-giá trị x trong mp1

    • cnt:=0

    • cnt:=giá trị nhỏ nhất của x và mp2 [khóa của x]

    • nếu cnt> 0, thì -

      • chèn khóa của x vào cuối res

  • trả lại res

Ví dụ

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -

#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));
}

Đầu vào

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

Đầu ra

[9, 6, ]