当前位置: 代码迷 >> 综合 >> 力扣(leetcode) 1442. 形成两个异或相等数组的三元组数目(前缀和匹配)
  详细解决方案

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

热度:58   发布时间:2023-12-26 11:50:51.0

题目: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 ^= ipreXOR.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 ^= ipreXOR.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)