统计异或值在范围内的数对有多少

题目描述

给你一个整数数组 nums （下标 从 0 开始 计数）以及两个整数：low 和 high ，请返回 漂亮数对 的数目。

漂亮数对 是一个形如 (i, j) 的数对，其中 0 <= i < j < nums.length 且 low <= (nums[i] XOR nums[j]) <= high 。

思路分析

• 利用字典树即可
• 从高位到低位进行比较并统计
• 位存储压缩到16位，比较从每次2*10^4数量级，压缩为2*O(10)以内(因为要进行两次比较)

代码如下

• 在大神基础上改进

class Trie {
public:
    Trie* next[2]={nullptr};
    int size=0;
    void insert(int num) {
        Trie* cur = this;

        for (int i = 15; i >= 0; i--) {
            int val = (num >> i) & 1;

            if (cur->next[val] == nullptr)
                cur->next[val] = new Trie();
            cur = cur->next[val];
            cur->size++;
        }
    }
    int search(int num, int limit) {
        Trie* cur = this;

        int res = 0;
        for (int i = 15; i >= 0; i--) {
            if (cur == nullptr) return res;
            int val = (num >> i) & 1;
            int vallim = (limit >> i) & 1;  //根据数据的大小，最高16位

            if (vallim == 1) {
                if (cur->next[val] != nullptr)
                    //因为要保证比较的一致性，一个为0，一个为1才有比较的可能
                    res += cur->next[val]->size;  //相同值的情况下，异或值就为0了
                cur = cur->next[1- val];
            }
            else cur = cur->next[val]; //当前位也要为0
        }
        return res;
    }
};

class Solution {
public:

    int countPairs(vector<int>& nums, int low, int high) {
        int res = 0;
        Trie *root=new Trie();
        for (auto num : nums) {
            res += root->search(num, high + 1)-root->search(num, low);
            root->insert(num);
        }
        return res;
    }
};