Codeforces 665E Beautiful Subarrays【01trie树】

题目链接：

http://codeforces.com/contest/665/problem/E

题意：

求异或值大于给定K的区间个数。

分析：

首先我们可以得到区间前缀的异或值。
这样我们将这个前缀M和K一起走trie树，如果该位K的值为0，那么无论怎么走最后得到的答案都不会比K小，所以直接加上另一边的子树大小，然后继续沿着当前边走。如果该位K的值为1，那么想要大于等于K必须沿着另一边贪心的走。

代码：

#include<cstdio>#include<iostream>using namespace std;const int maxn = 1000000 << 5, maxm = 1e6 + 5;int f[maxn][2];int cnt = 1;int son[maxn], pre[maxm];long long ans = 0;#define pl(n) cout<<#n<<": "<<n<<endl;#define sa(n) scanf("%d", &n)void insert(int n){    int k;    int u = 1;    for(int i = 30; i >= 0; i--){        k = (n >> i) & 1;        if(!f[u][k]) f[u][k] = ++cnt;        son[u]++;        u = f[u][k];    }    son[u]++;}void query(int n, int m){    int k1, k2, u = 1;    for(int i = 30; i >= 0; i--){        k1 = (n >> i) & 1;        k2 = (m >> i) & 1;        if(k2 == 0){            ans += son[f[u][!k1]];            u = f[u][k1];        }        else u = f[u][!k1];        if(u == 0) break;    }    ans += son[u];}int main (void){    int n,k;sa(n);sa(k);    pre[0] = 0;    int num;    insert(pre[0]);    for(int i = 1; i <= n; i++){        sa(num);        pre[i] = pre[i - 1] ^ num;        query(pre[i], k);        insert(pre[i]);        //pl(ans);    }    printf("%I64d\n", ans);    return 0;}