Leetcode 421. 数组中两个数的最大异或值 C++

Leetcode 421. 数组中两个数的最大异或值

题目

给定一个非空数组,数组中元素为 a0, a1, a2, … , an-1,其中 0 ≤ ai < 231

找到 ai 和aj 最大的异或 (XOR) 运算结果,其中0 ≤ i, j < n 。

你能在O(n)的时间解决这个问题吗?

示例:

输入: [3, 10, 5, 25, 2, 8]

输出: 28

解释: 最大的结果是 5 ^ 25 = 28.

测试样例

前缀树
我们可以根据每个数位建立一个前缀数。对于当前的数,从高位到低位,我们尽可能的遍历和自己数位不同的结点,显然就是得到当前数异或的最大值。详细过程见代码

代码

struct Tree{
    Tree* zero;
    Tree* one;
    int value;
    Tree(){
        zero = NULL;
        one = NULL;
    }
};
class Solution {
public:
    int findMaximumXOR(vector<int>& nums) {
        int n = nums.size();
        Tree* root = new Tree();
        for(int i=0; i<n; i++){		//建立前缀树
            Tree* now = root;
            for(int j=31; j>=0; j--){
                int cur = ((nums[i])>>j) & 1;
                if(cur == 0){
                    if(now->zero == NULL){
                        Tree* newT = new Tree();
                        now->zero = newT;
                    }
                    now = now->zero;
                }else{
                    if(now->one == NULL){
                        Tree* newT = new Tree();
                        now->one = newT;
                    }
                    now = now->one;
                }
            }
            now->value = nums[i];
        }

        int ans=0;
        for(int i=0; i<n; i++){
            Tree* now = root;
            for(int j=31; j>=0; j--){
                int cur = ((nums[i])>>j) & 1;
                if(cur==0){		//当前位为0,我们应该遍历1
                    if(now->one)    now = now->one;	//存在1就遍历
                    else    now = now->zero;
                }else{			//当前位为1,我们应该遍历0
                    if(now->zero)   now = now->zero;	//有0就遍历
                    else    now = now->one;
                }
            }
            ans = max(ans,nums[i]^now->value);
        }
        return ans;
    }
};

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-xor-of-two-numbers-in-an-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

你可能感兴趣的:(Leetcode 421. 数组中两个数的最大异或值 C++)