arrays - 找出数组中三元组i，j，k的数量，以使索引为i到j-1的元素的异或等于索引为j到k的元素的异或

对于给定的正整数序列a1，a2，…，an，你应该找到三元组的数目（i，j，k），这样ai^ai+1^..^aj-1=aj^aj+1^..ak
其中，^表示按位异或。
问题的链接在这里：https://www.codechef.com/AUG19B/problems/KS1
我所做的就是试图找到所有xor为0的子阵。解是二次的，所以太慢了。
这是我设法找到的解决办法。

for (int i = 0; i < arr.length; i++) {
            int xor = arr[i];
            for (int j = i + 1; j < arr.length; j++) {
                xor ^= arr[j];
                if (xor == 0) {
                    ans += (j - i);
                }
            }
        }
        finAns.append(ans + "\n");

最佳答案

以下是根据Ciapan在问题描述下的评论得出的O(n)解决方案：
如果索引i到j-1处的项的异或等于从j到k的异或，则从i到k的异或等于零。对于任何这样的子阵k]i+1和k-1之间的每一个j构成一个满足要求的三元组。从i到k的xor等于（从0到k的xor）xor（从0到i-1的xor）。所以我想您可能会找到序列所有可能的初始部分的xor-s，并寻找它们的相等对。
功能f是主要方法brute_force用于验证。
Python2.7代码：

import random

def brute_force(A):
  res = 0

  for i in xrange(len(A) - 1):
    left = A[i]
    for j in xrange(i + 1, len(A)):
      if j > i + 1:
        left ^= A[j - 1]
      right = A[j]
      for k in xrange(j, len(A)):
        if k > j:
          right ^= A[k]
        if left == right:
          res += 1

  return res

def f(A):
  ps = [A[0]] + [0] * (len(A) - 1)
  for i in xrange(1, len(A)):
    ps[i] = ps[i- 1] ^ A[i]

  res = 0
  seen = {0: (-1, 1, 0)}

  for  i in xrange(len(A)):
    if ps[i] in seen:
      prev_i, i_count, count = seen[ps[i]]
      new_count = count + i_count * (i - prev_i) - 1
      res += new_count
      seen[ps[i]] = (i, i_count + 1, new_count)
    else:
      seen[ps[i]] = (i, 1, 0)

  return res

for i in xrange(100):
  A = [random.randint(1, 10) for x in xrange(200)]
  f_A, brute_force_A = f(A), brute_force(A)
  assert f_A == brute_force_A
print "Done"

关于arrays - 找出数组中三元组i，j，k的数量，以使索引为i到j-1的元素的异或等于索引为j到k的元素的异或，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/57839761/