题目: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)