C++中统计可构成两个异或值相等的数组的三元组
假设我们有一个整数数组arr。我们想要选择三个索引i、j和k,其中(0 <= i < j <= k < N),N是数组的大小。a和b的值如下:a = arr[i] XOR arr[i + 1] XOR ... XOR arr[j - 1] b = arr[j] XOR arr[j + 1] XOR ... XOR arr[k] 我们需要找到a等于b的三元组(i, j, k) 的数量。
因此,如果输入是[2,3,1,6,7],则输出将是4,因为三元组是(0,1,2)、(0,2,2)、(2,3,4)和(2,4,4)
为了解决这个问题,我们将遵循以下步骤:
ret := 0
n := arr的大小
for i := 1 to n-1 do −
定义一个map m
x1 := 0, x2 := 0
for j := i - 1 downto 0 do −
x1 := x1 XOR arr[j]
m[x1] := m[x1] + 1
for j := i to n-1 do −
x2 := x2 XOR arr[j]
ret := ret + m[x2]
return ret
示例
让我们来看下面的实现以更好地理解:
#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
广告