力扣(leetcode) 1442. 形成两个异或相等数组的三元组数目(前缀和匹配)

题目:https://leetcode-cn.com/problems/count-triplets-that-can-form-two-arrays-of-equal-xor/

题目比较容易理解，直接能想到的就是把i,j,k全部列出来，然后暴力循环求a,b 找答案。

不过这种方法提交之后必然超时。优化一下…
加入之前用过的前缀异或和方法 ,但i,j,k依旧暴力三层循环解.

arr = [2,3,1,6,7]
preXOR = [] #存前缀异或数组
res = 0
temp = 0
for i in arr:
    temp ^= i
    preXOR.append(temp)
for i in range(len(arr)):
    for j in range(i+1,len(arr)):
    # 注意 这个题要求是i到j-1 和j到k
    # 所以这里从i+1 而下面是从j开始 
        for k in range(j,len(arr)):
                print(i,j,k)
                a = preXOR[i] ^ preXOR[j-1] ^ arr[i]
                b = preXOR[j] ^ preXOR[k] ^ arr[j]
                if a==b :
                    res +=1
print(res)

上面的代码放到LeetCode上可以提交通过。

进一步优化为两层for：
仔细读题 求当满足a=b时候的三元组。
而
a = arr[i] ^ arr[i + 1] ^ … ^ arr[j - 1]
b = arr[j] ^ arr[j + 1] ^ … ^ arr[k]
所以 a=b 即
arr[i] ^ arr[i + 1] ^ … ^ arr[j - 1] = arr[j] ^ arr[j + 1] ^ … ^ arr[k]

且我们知道 两个相等的数异或时 等于0

也就是说arr从i项到第k项的异或值等于0
所以在这时候 我们只需要列举出i和k的位置就可以了
j无论在哪 都不影响结果 因为当a=b的时候 中间那一块都等于0

也就是说 当前缀异或数组前i-1项和前k项相等 就满足条件（i-k项异或为0）

arr = [1,3,5,7,9]
preXOR = []
res = 0
temp = 0
# preXOR  前缀异或和数组
for i in arr:
    temp ^= i
    preXOR.append(temp)
print(preXOR)
for i in range(len(arr)):
        for k in range(i+1,len(arr)):
                if preXOR[i] ^ arr[i] == preXOR[k]:
                # 这里按照理解 
                #  应该是preXOR[i-1]  == preXOR[k]
                # 但是当第一次循环 i为0的时候 i-1下标越界了
                # 所以直接改为  preXOR[i] ^ arr[i]
                # ps: preXOR[i] ^ arr[i] = preXOR[i-1]
                    res +=k-i
                    # 这里别写成 res += 1  
                    # 因为我们省略了中间的i 所以所有的可能为k-i
                    # k-i中间有几个数 都可能是j  
print(res)