JAVA程序设计：统计异或值在范围内的数对有多少（LeetCode：1803）

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

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

 

示例 1：

输入：nums = [1,4,2,7], low = 2, high = 6
输出：6
解释：所有漂亮数对 (i, j) 列出如下：
    - (0, 1): nums[0] XOR nums[1] = 5 
    - (0, 2): nums[0] XOR nums[2] = 3
    - (0, 3): nums[0] XOR nums[3] = 6
    - (1, 2): nums[1] XOR nums[2] = 6
    - (1, 3): nums[1] XOR nums[3] = 3
    - (2, 3): nums[2] XOR nums[3] = 5
示例 2：

输入：nums = [9,8,4,2,1], low = 5, high = 14
输出：8
解释：所有漂亮数对 (i, j) 列出如下：
​​​​​    - (0, 2): nums[0] XOR nums[2] = 13
    - (0, 3): nums[0] XOR nums[3] = 11
    - (0, 4): nums[0] XOR nums[4] = 8
    - (1, 2): nums[1] XOR nums[2] = 12
    - (1, 3): nums[1] XOR nums[3] = 10
    - (1, 4): nums[1] XOR nums[4] = 9
    - (2, 3): nums[2] XOR nums[3] = 6
    - (2, 4): nums[2] XOR nums[4] = 5
 

提示：

1 <= nums.length <= 2 * 104
1 <= nums[i] <= 2 * 104
1 <= low <= high <= 2 * 104

思路：我们考虑01字典树，对于每个数，我们考虑它和已插入的所有数所能异或得到的在[low,high]区间内的数量，这是字典树可以做到的！

class Solution {

    class node {
        int num;
        node[] nxt;

        public node(int num) {
            this.num = num;
            nxt = new node[2];
        }
    }

    private node root;

    public int countPairs(int[] nums, int low, int high) {
        int ans = 0;
        root = new node(0);

        for (int num : nums) {
            ans += query(num, high + 1) - query(num, low);
            insert(num);
        }
        return ans;
    }

    private int query(int x, int high) {
        int res = 0;
        node now = root;
        for (int i = 16; i >= 0; i--) {
            int u = (x >> i) & 1, v = (high >> i) & 1;
            if (now == null)
                break;
            if (v == 1) {
                res += now.nxt[u] == null ? 0 : now.nxt[u].num;
                now = now.nxt[1 - u];
            } else
                now = now.nxt[u];
        }
        return res;
    }

    private void insert(int x) {
        node now = root;
        for (int i = 16; i >= 0; i--) {
            int t = (x >> i) & 1;
            if (now.nxt[t] == null)
                now.nxt[t] = new node(0);
            now = now.nxt[t];
            now.num++;
        }
    }
}