LC421.數組中兩個數的最大異或結果-Trie樹

題目描述

解題思路

總的來說就只需要兩步

1. 將數組中的數全部存入字典樹中
2. 遍歷樹中的每一個數在字典樹中異或的最大結果，最後再求最大結果裏面的最大值返回就是了

class Trie{
public:
	// map<int, vector<int>> next;
    Trie* next[2];
    Trie()
    {
        memset(next, 0, sizeof(next));
    }
};
class Solution {
    Trie* root = new Trie();
public:
    int findMaximumXOR(vector<int>& nums) {
        // 將數全部存入字典樹裏面
        for(int num : nums)
        {
            Trie* node = root;
            for(int i = 30; i >= 0; i--)
            {
                int bt = num >> i & 1;
                if(node->next[bt] == nullptr)
                {
                    node->next[bt] = new Trie();
                }
                node = node->next[bt];
            }
        }
        // 找最大^值
        int res = 0;
        for(int num : nums)
        {
            Trie* node = root;
            int sum = 0;
            for(int i = 30; i >= 0; i--)
            {
                int bt = num >> i & 1;
                if(bt == 1)
                {
                    sum += node->next[0] != nullptr ? 1 << i : 0 ;
                    node = node->next[0] != nullptr ? node->next[0] : node->next[1];
                }
                else
                {
                    sum += node->next[1] != nullptr ? 1 << i : 0 ;
                    node = node->next[1] != nullptr ? node->next[1] : node->next[0];
                }
            }
            res = max(res, sum);
        }
        return res;
    }
};

時間複雜度：O（N）
空間複雜度：O（1）