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;
}