Codeforces Round #340 (Div. 2) E. XOR and Favorite Number（莫队算法）

最新推荐文章于 2018-09-11 15:43:21 发布

TaoSama

最新推荐文章于 2018-09-11 15:43:21 发布

阅读量402

点赞数

分类专栏：数据结构文章标签：莫队算法

本文链接：https://blog.csdn.net/lwt36/article/details/50583802

版权

数据结构专栏收录该内容

21 篇文章 0 订阅

订阅专栏

博客介绍了如何使用莫队算法解决一道关于XOR操作和区间内满足特定异或和条件的询问问题。通过维护异或前缀和，将原问题转化为求解满足特定异或和的子区间对数量，并采用离线分块策略进行高效处理。

摘要由CSDN通过智能技术生成

题意:

$给定N\le 10^5个数的序列, Q\le 10^5次询问, 给定一个数K\le 10^6$
$询问[L,R] 中, 满足i\le j,i,j\in[L,R], 且a_i\oplus a_{i+1}\oplus\cdots\oplus a_j=K, (i,j)对有多少个$

分析:

$维护异或前缀和, prefix_i=prefix_{i-1}\oplus a_i, prefix_0=0$
$那么a_l\oplus a_{l+1}\oplus\cdots\oplus a_r = prefix_{l-1}\oplus prefix_r$
$问题则转变成了: 满足i\le j,i,j\in[L-1,R],prefix_i\oplus prefix_j=K的(i,j)对的个数$
$对询问进行离线分块即可使用莫队算法做了$

代码:

//
//  Created by TaoSama on 2016-01-25
//  Copyright (c) 2015 TaoSama. All rights reserved.
//
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <algorithm>
#include <cctype>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <string>
#include <set>
#include <vector>

using namespace std;
#define pr(x) cout << #x << " = " << x << "  "
#define prln(x) cout << #x << " = " << x << endl
const int N = 1e5 + 10, INF = 0x3f3f3f3f, MOD = 1e9 + 7;
const int B = 400;

int n, q, k;
int pre[N], cnt[1 << 20];

struct Query {
    int l, r, id;
    bool operator<(const Query& q) const {
        return r < q.r;
    }
};

long long sum, ans[N];
void update(int i, int delta) {
    if(delta > 0) {
        sum += cnt[pre[i] ^ k];
        ++cnt[pre[i]];
    } else {
        --cnt[pre[i]];
        sum -= cnt[pre[i] ^ k];
    }
}

int main() {
#ifdef LOCAL
    freopen("C:\\Users\\TaoSama\\Desktop\\in.txt", "r", stdin);
//  freopen("C:\\Users\\TaoSama\\Desktop\\out.txt","w",stdout);
#endif
    ios_base::sync_with_stdio(0);

    while(scanf("%d%d%d", &n, &q, &k) == 3) {
        for(int i = 1; i <= n; ++i) {
            scanf("%d", pre + i);
            pre[i] ^= pre[i - 1];
        }
        memset(cnt, 0, sizeof cnt);
        vector<vector<Query> > Q(n / B + 2, vector<Query>());
        for(int i = 1; i <= q; ++i) {
            int l, r; scanf("%d%d", &l, &r);
            Q[l / B].push_back(Query {--l, ++r, i});
        }
        for(auto &i : Q) sort(i.begin(), i.end());

        sum = 0;
        for(int i = 0; i < Q.size(); ++i) {
            int l, r; l = r = i * B;
            for(auto &q : Q[i]) {
                while(r < q.r) update(r++, 1);
                while(l < q.l) update(l++, -1);
                while(l > q.l) update(--l, 1);
                ans[q.id] = sum;
            }
            for(int j = l; j < r; ++j) update(j, -1);
        }

        for(int i = 1; i <= q; ++i) printf("%I64d\n", ans[i]);
    }
    return 0;
}