1. 程式人生 > 其它 >CodeForces-617E XOR and Favorite Number

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