leetcode刷题笔记——421. 数组中两个数的最大异或值

最新推荐文章于 2022-04-05 18:24:15 发布

高木同学丶

最新推荐文章于 2022-04-05 18:24:15 发布

阅读量150

点赞数

分类专栏： LeetCode 文章标签：数据结构

本文链接：https://blog.csdn.net/weixin_43904469/article/details/117191074

版权

LeetCode 专栏收录该内容

20 篇文章 0 订阅

订阅专栏

该博客介绍了一种使用字典树（Trie）数据结构来高效寻找给定整数集合中与目标数异或结果最大的值的方法。通过遍历字典树并根据异或运算的特性，可以实现O(log n)的时间复杂度。博客内容包括字典树的定义、添加元素到树中以及获取最大异或值的函数实现，并提供了一个完整的C++代码示例。

摘要由CSDN通过智能技术生成

题目：

在这里插入图片描述

思路：

采用字典树，所谓字典树见下图
在这里插入图片描述
从高位保存数字，将数字的二进制形式的最高位保存至树的最高层。
这样我们可以根据异或的性质，只有两个不同的值才能得1，也就是说，当要获取树中与0010异或结果最大的数，优先考虑的是1101，每一位都不同，此处采用贪心思想。如果树中不存在与某位相反的值，则取相同值。

采用字典树的好处是，仅需要log(n)就可以得到能够取得最大异或值的数。

字典树的数据结构：

struct Trie{
    Trie* left = nullptr;       //字典树左指针，用于存储0
    Trie* right = nullptr;      //字典树右指针，用于存储1

    Trie(){}	//struct的构造函数
};

往字典树中添加元素：

    //往字典树中添加元素
    void add(int num,int HIGHBIT){
        Trie* cur = root;
        for(int i=HIGHBIT;i>=0;i--){
            int bit = (num>>i)&1; //取num的第i位
            if(bit == 0){           //第i位为0，创建左孩子
                if(!cur->left){
                    cur->left = new Trie();
                }
                cur = cur->left;
            }else{                  //第i位为1，创建右孩子
                if(!cur->right){
                    cur->right = new Trie();
                }
                cur = cur->right;
            }
        }
    }

得到与num异或结果的最大值：

    //取出当前字典树中与num异或结果的最大值
    int check(int num,int HIGHBIT){
        Trie* cur = root;
        int x=0;
        for(int i=HIGHBIT;i>=0;i--){
            int bit = (num>>i)&1;
            if(bit == 0){
                if(cur->right){
                    cur = cur->right;       //num的第i位是0，理论上获得最大的异或值应该指针往右走
                    x=x*2+1;
                }else{
                    cur = cur->left;
                    x=x*2;
                }
            }else{                          //num的第i位是1，理论上获得最大的异或值应该指针往左走
                if(cur->left){				//num的第i位是1，指针若能往左走说明树中存在此位为0的数，则异或结果的此位值为1
                    cur = cur->left;
                    x=x*2+1;
                }else{
                    cur = cur->right;
                    x=x*2;
                }
            }
        }
        return x;
    }

完整代码：

struct Trie{
    Trie* left = nullptr;       //字典树左指针，用于存储0
    Trie* right = nullptr;      //字典树右指针，用于存储1

    Trie(){}
};

class Solution {
private:
    Trie* root = new Trie();

public:
    //往字典树中添加元素
    void add(int num,int HIGHBIT){
        Trie* cur = root;
        for(int i=HIGHBIT;i>=0;i--){
            int bit = (num>>i)&1; //取num的第i位
            if(bit == 0){           //第i位为0，创建左孩子
                if(!cur->left){
                    cur->left = new Trie();
                }
                cur = cur->left;
            }else{                  //第i位为1，创建右孩子
                if(!cur->right){
                    cur->right = new Trie();
                }
                cur = cur->right;
            }
        }
    }
   //取出当前字典树中与num异或结果的最大值
    int check(int num,int HIGHBIT){
        Trie* cur = root;
        int x=0;
        for(int i=HIGHBIT;i>=0;i--){
            int bit = (num>>i)&1;
            if(bit == 0){
                if(cur->right){
                    cur = cur->right;       //num的第i位是0，理论上获得最大的异或值应该指针往右走
                    x=x*2+1;
                }else{
                    cur = cur->left;
                    x=x*2;
                }
            }else{                          //num的第i位是1，理论上获得最大的异或值应该指针往左走
                if(cur->left){
                    cur = cur->left;
                    x=x*2+1;
                }else{
                    cur = cur->right;
                    x=x*2;
                }
            }
        }
        return x;
    }
    int findMaximumXOR(vector<int>& nums) {
        int n = nums.size();
        int ans = 0;
        int maxnums = *max_element(nums.begin(), nums.end());
        if(maxnums==0)return 0;
        int HIGHBIT = ceil(log(maxnums)/log(2));
        for(int i=0;i<n;i++){
            add(nums[i],HIGHBIT);
            ans = max(ans,check(nums[i],HIGHBIT));
        }
        return ans;
    }
};