Edu12 Beautiful Subarrays --- 题解

时间:2024-03-17 16:07:03

 Beautiful Subarrays:

题目大意:

  思路解析:

要找到一个区间并且区间的l--r里面所有的元素异或值大于等于k,称这样的数组是优美子数组,问优美子数组有多少个。

[L,R] 的数组异或和等价于 (a1,a2,a3,....aL-1) ^ (a1,a2,a3,a4,....aR)那么发现我们可以用前缀和的思想,来找到一个前缀使得[L,R]是优美子数组,发现这种二进制的寻找可以利用字典树实现,然后每次在这样的字典树插入前缀。

寻找时,如果加上当前位就能大于等于k,那么我们并不用关心以后的情况了,直接加上满足能加上当前位的所有情况即可,这里需要利用一个标记位进行预处理。

代码实现:


#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 3e7 + 5;
int ch[maxn][2];
int tag[maxn];
int tot = 0;
ll ans = 0;
int k;
void insert(int x){
    int p = 0;
    tag[p]++;
    for (int i = 30; i >= 0; i--) {
        int c = (x >> i) & 1;
        if (ch[p][c] == 0) ch[p][c] = ++tot;
        p = ch[p][c];
        tag[p]++;
    }
}

void get(int x){
    int cur = 0;
    int p = 0;
    for (int j = 30; j >= 0; j--) {

        int c = (x >> j) & 1;
        if ((cur | (1 << j)) >= k){
            ans += ch[p][c ^ 1] != 0 ? tag[ch[p][c ^ 1]] : 0;
            p = ch[p][c];
        }else {
            cur |= (1 << j);
            p = ch[p][c ^ 1];
        }
        if (p == 0) break;
    }
    if (cur >= k) ans += p != 0 ?tag[p]:0;
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    memset(ch, 0, sizeof(ch));
    memset(tag, 0, sizeof(ch));
    int n;
    cin >> n >> k;
    int s = 0;
    insert(s);
    for(int i = 0; i < n; ++i){
        int num;
        cin >> num;
        s ^= num;
        get(s);
        insert(s);
    }
    cout << ans << endl;
    return 0;
}