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

C++에서 동일한 XOR의 두 배열을 형성할 수 있는 삼중항 계산

<시간/>

정수 배열이 있다고 가정합니다. i, j 및 k와 같은 세 가지 인덱스를 선택하려고 합니다. 여기서 (0 <=i

따라서 입력이 [2,3,1,6,7]과 같으면 삼중항이 (0,1,2), (0,2,2), (2,3)이므로 출력은 4가 됩니다. ,4) 및 (2,4,4)

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

  • 렛 :=0

  • n :=arr의 크기

  • initialize i :=1의 경우, i

    • 하나의 맵 정의

    • x1 :=0, x2 :=0

    • 초기화 j :=i - 1의 경우 j>=0일 때 업데이트(j를 1만큼 감소), −

      • x1 :=x1 XOR arr[j]

      • (m[x1] 1씩 증가)

    • 초기화 j :=i의 경우 j

      • x2 :=x2 XOR arr[j]

      • 렛 :=렛 + m[x2]

  • 리턴 렛

예시

이해를 돕기 위해 다음 구현을 살펴보겠습니다. −

#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
   int countTriplets(vector<int>& arr) {
      int ret = 0;
      int n = arr.size();
      for (int i = 1; i < n; i++) {
         map<int, int> m;
         int x1 = 0;
         int x2 = 0;
         for (int j = i - 1; j >= 0; j--) {
            x1 = x1 ^ arr[j];
            m[x1]++;
         }
         for (int j = i; j < n; j++) {
            x2 = x2 ^ arr[j];
            ret += m[x2];
         }
      }
      return ret;
   }
};
main(){
   Solution ob;
   vector<int> v = {2,3,1,6,7};
   cout << (ob.countTriplets(v));
}

입력

{2,3,1,6,7}

출력

4