给定一个数组arr，想知道arr中哪两个数的异或结果最大。返回最大的异或结果。
前缀树。一个数，用二进制表示，0走左边分支，1走右边分支。准备一个max变量，遍历的时候，遇到比max还要大的，max更新。最后返回max。
时间复杂度：O(N)。
代码用golang编写。代码如下：
package main
import (
"fmt"
"math"
)
func main() {
arr := []int{3, 10, 5, 25, 2, 8}
ret := findMaximumXOR(arr)
fmt.Println(ret)
}
func findMaximumXOR(arr []int) int {
if len(arr) < 2 {
return 0
}
max := math.MinInt64
numTrie := NewNumTrie()
numTrie.add(arr[0])
for i := 1; i < len(arr); i++ {
max = getMax(max, numTrie.maxXor(arr[i]))
numTrie.add(arr[i])
}
return max
}
type Node struct {
nexts []*Node
}
func NewNode() *Node {
ret := &Node{}
ret.nexts = make([]*Node, 2)
return ret
}
// 基于本题，定制前缀树的实现
type NumTrie struct {
// 头节点
head *Node
}
func NewNumTrie() *NumTrie {
ret := &NumTrie{}
ret.head = NewNode()
return ret
}
func (this *NumTrie) add(newNum int) {
cur := this.head
for move := 63; move >= 0; move-- {
path := (newNum >> move) & 1
if cur.nexts[path] == nil {
cur.nexts[path] = NewNode()
}
cur = cur.nexts[path]
}
}
// 该结构之前收集了一票数字，并且建好了前缀树
// num和 谁 ^ 最大的结果（把结果返回）
func (this *NumTrie) maxXor(num int) int {
cur := this.head
ans := 0
for move := 63; move >= 0; move-- {
// 取出num中第move位的状态，path只有两种值0就1，整数
path := (num >> move) & 1
// 期待遇到的东西
best := twoSelectOne(move == 63, path, path^1)
// 实际遇到的东西
best = twoSelectOne(cur.nexts[best] != nil, best, best^1)
// (path ^ best) 当前位位异或完的结果
ans |= (path ^ best) << move
cur = cur.nexts[best]
}
return ans
}
func twoSelectOne(condition bool, a int, b int) int {
if condition {
return a
} else {
return b
}
}
func getMax(a int, b int) int {
if a > b {
return a
} else {
return b
}
}执行结果如下：
