C语言 | Leetcode C语言题解之第421题数组中两个数的最大异或值

题目:

题解:

cpp 复制代码
const int HIGH_BIT = 30;

struct Trie {
    // 左子树指向表示 0 的子节点
    struct Trie* left;
    // 右子树指向表示 1 的子节点
    struct Trie* right;
};

struct Trie* createTrie() {
    struct Trie* ret = malloc(sizeof(struct Trie));
    ret->left = ret->right = NULL;
    return ret;
}

void add(struct Trie* root, int num) {
    struct Trie* cur = root;
    for (int k = HIGH_BIT; k >= 0; --k) {
        int bit = (num >> k) & 1;
        if (bit == 0) {
            if (!cur->left) {
                cur->left = createTrie();
            }
            cur = cur->left;
        } else {
            if (!cur->right) {
                cur->right = createTrie();
            }
            cur = cur->right;
        }
    }
}

int check(struct Trie* root, int num) {
    struct Trie* cur = root;
    int x = 0;
    for (int k = HIGH_BIT; k >= 0; --k) {
        int bit = (num >> k) & 1;
        if (bit == 0) {
            // a_i 的第 k 个二进制位为 0,应当往表示 1 的子节点 right 走
            if (cur->right) {
                cur = cur->right;
                x = x * 2 + 1;
            } else {
                cur = cur->left;
                x = x * 2;
            }
        } else {
            // a_i 的第 k 个二进制位为 1,应当往表示 0 的子节点 left 走
            if (cur->left) {
                cur = cur->left;
                x = x * 2 + 1;
            } else {
                cur = cur->right;
                x = x * 2;
            }
        }
    }
    return x;
}

int findMaximumXOR(int* nums, int numsSize) {
    struct Trie* root = createTrie();
    int x = 0;
    for (int i = 1; i < numsSize; ++i) {
        // 将 nums[i-1] 放入字典树,此时 nums[0 .. i-1] 都在字典树中
        add(root, nums[i - 1]);
        // 将 nums[i] 看作 ai,找出最大的 x 更新答案
        x = fmax(x, check(root, nums[i]));
    }
    return x;
}
相关推荐
灵哎惹,凌沃敏32 分钟前
C语言/Keil的register修饰符
c语言·开发语言
皮蛋sol周36 分钟前
嵌入式学习C语言(八)二维数组及排序算法
c语言·学习·算法·排序算法
森焱森1 小时前
单片机中 main() 函数无 while 循环的后果及应对策略
c语言·单片机·算法·架构·无人机
IT永勇2 小时前
数据结构-栈
c语言·数据结构·嵌入式开发
定偶2 小时前
进制转换小题
c语言·开发语言·数据结构·算法
森焱森3 小时前
60 美元玩转 Li-Fi —— 开源 OpenVLC 平台入门(附 BeagleBone Black 驱动简单解析)
c语言·单片机·算法·架构·开源
千帐灯无此声3 小时前
Linux 测开:日志分析 + 定位 Bug
linux·c语言·c++·bug
ChuHsiang4 小时前
【C语言学习】实现游戏——三子棋
c语言
科大饭桶5 小时前
数据结构自学Day5--链表知识总结
数据结构·算法·leetcode·链表·c