给你一个整数数组 nums ,返回 nums[i] XOR nums[j] 的最大运算结果,其中 0 ≤ i ≤ j < n 。
进阶:你可以在 O(n) 的时间解决这个问题吗?
如果没有限定条件 O(n) 那么我就可以使用暴力法遍历两遍,将所有数都与其他数进行异或操作,就可以计算出最大异或值。
但是有限定条件的话,感觉就要使用一些数据结构了,依靠我当前的水平还是做不出这种题,学习了一遍,关于构造树以及前缀树等知识。
class Trie:
def __init__(self, val):
self.val = val
self.child = {}
class Solution:
def findMaximumXOR(self, nums: List[int]) -> int:
root = Trie(-1)
L = len(format(max(nums), 'b')) - 1
for n in nums:
curr = root
for i in range(L, -1, -1):
tem = (n >> i) & 1
if tem not in curr.child:
curr.child[tem] = Trie(tem)
curr = curr.child[tem]
res = 0
for n in nums:
total = 0
curr = root
for i in range(L, -1, -1):
tem = (n >> i) & 1
if 1-tem in curr.child:
total = total*2 + 1
curr = curr.child[1-tem]
else:
total *= 2
curr = curr.child[tem]
res = max(res, total)
return res
首先需要构造一棵树,遍历 nums 里所有的数,将其构造成一棵二进制树,树的深度是 nums 数组中最大数的二进制长度,数的在该位的值为 0 时构造在左叶子,为 1 时构造为右叶子;
再次遍历数组,计算最大异或值,使用贪心,一直致力于查找与其位相反的数,比如该数在当前位为 1 ,那么它就先寻找为 0 的数,没有再考虑为 1 的数,这样才能保证异或出的结果最大。