力扣周赛D-统计异或值范围-字典树

前言:

比赛的时候已经确定是字典树了.细节没想清楚.算是一道模板题吧,但是我还没写出来(

题目大意:

给你一个序列$a_i$.问你有多少个数对$(i,j)$使得$l\le a_i\oplus a_j\le h$.

题目思路:

分成两段求.问题变成考虑枚举每一个$a_i$,如何快速求序列中有多少个数异或它 $\le x$.

对序列建立字典树。在字典树上搜索。

具体做法：

从最高位开始考虑:设当前数为$a_i$,当前在第$j$位，上界是$lim$.

如果我们在第$j$位填写$!a_{i,j}$.那么这一位会产生$2^j$的贡献。分两种情况：

$1.lim<2^j$,那么现在不能填不同的数，必须填相同的数。那么递归往$a_{i,j}$走。

$2.lim\ge 2^j$,那么现在都可以填。

2.1 如果填相同的，那么二进制位$[0,j-1]$就可以随便填了。直接加上子树sz.

2.2 如果填不同的，那么再递归求解。

递归出口：到叶子节点后返回sz.

合理性：

其实这个过程是枚举$lim$的所有$1$的位置进行计算的。

因为任意一个数$x\le lim$,从二进制的角度来看一定 是$x$和$lim$有一段前缀是相同的，直到某一位$li{m}_i=1$但$x_i=0$.

那计算小于等于$lim$的个数就是枚举每一个$li{m}_i=1$的位置,前缀保证一样，计算所有后继的个数。这个过程就在字典树上$dfs$来完成即可。

复杂度：

显然，复杂度为:$O(nlog(ma{x}_a))$

AC代码:

#define ll long long
#define pii pair<int,int>
#define pb push_back
#define mp make_pair
#define vi vector<int>
#define vll vector<ll>
#define fi first
#define se second
const int maxn = 20005 * 15 + 5;
const int mod = 1e9 + 7;
class Solution {
public:
    int tr[maxn][2] , sz[maxn] , cnt;
    void add (int x){
        int now = 0;
        sz[now]++;
        for (int i = 16 ; i >= 0 ; i--){
            int d = ((x >> i) & 1);
            if (!tr[now][d]) tr[now][d] = ++cnt;
            now = tr[now][d];
            sz[now]++;
        }
    }
    ll dfs (int id , int dep , int val , int lim) {
        if (dep == -1) return sz[id];
        int d = (val >> dep) & 1 , p = (1 << dep);
        ll ans = 0;
        if (lim < p) {
            if (tr[id][d]) ans += dfs(tr[id][d] , dep - 1 , val , lim);
        }else {
            if (tr[id][d]) ans += sz[tr[id][d]];
            if (tr[id][!d]) ans += dfs(tr[id][!d] , dep - 1 , val , lim - p);
        }
        return ans;
    }
    int countPairs(vector<int>& a, int x, int y) {
        ll ans = 0;
        int n = a.size();
        for (int i = 0 ; i < n ; i++){
            ll l = dfs(0 , 16 , a[i] , x - 1) , h = dfs(0 , 16 , a[i] , y);
            ans += h - l;add(a[i]);
        }
        return ans;
    }
};