Loading

CodeForces-617E XOR and Favorite Number

XOR and Favorite Number

询问一个区间,求所以异或值为 k 的子串数量

普通莫队

一开始一直在想这个异或怎么处理,一直卡在 \(O(n^2)\) 的想法,后来发现可以用前缀异或和,问题就转化成在 \([l-1,r]\) 区间中,找到两个前缀和,使得 \(sum_i \oplus sum_j = k\),直接桶维护一下就行

但是前缀和就是会错一位,特难受,所以直接把区间改成 \([l-1, r]\)

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 10;
int n, m, k;
int num[maxn], sum[maxn], cnt[maxn * 20], pos[maxn];
ll ans = 0, last[maxn];

struct node
{
    int l, r, id;
    node(){}
    node(int _l, int _r, int _id){l = _l; r = _r; id = _id;}
}seg[maxn];

bool cmp(const node& a, const node& b)
{
    return pos[a.l] ^ pos[b.l] ? pos[a.l] < pos[b.l] : (pos[a.l] & 1 ? a.r > b.r : a.r < b.r);
}

inline void add(int x)
{
    int now = sum[x];
    ans += cnt[k ^ now];
    cnt[now]++;
}

inline void del(int x)
{
    int now = sum[x];
    cnt[now]--;
    ans -= cnt[k ^ now];
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> m >> k;
    for(int i=1; i<=n; i++) cin >> num[i];
    for(int i=1; i<=n; i++) sum[i] = sum[i-1] ^ num[i];
    for(int i=0; i<m; i++)
    {
        int l, r;
        cin >> l >> r;
        l--;
        seg[i] = node(l, r, i);
    }
    int t = sqrt(n);
    for(int i=1; i<=n; i++) pos[i] = i / t;
    sort(seg, seg + m, cmp);
    int l = 2, r = 1;
    for(int i=0; i<m; i++)
    {
        while(r < seg[i].r) add(++r);
        while(l < seg[i].l) del(l++);
        while(l > seg[i].l) add(--l);
        while(r > seg[i].r) del(r--);
        last[seg[i].id] = ans;
    }
    for(int i=0; i<m; i++)
        cout << last[i] << endl;
    return 0;
}
posted @ 2022-05-27 11:02  dgsvygd  阅读(13)  评论(0编辑  收藏  举报