2018年第三阶段个人训练赛第九场6759 Problem A 异或序列

题目描述

已知一个长度为n的整数数列a1,a2,…,an，给定查询参数l、r，问在al,al+1,…,ar区间内，有多少子序列满足异或和等于k。也就是说，对于所有的x,y(l≤x≤y≤r)，满足ax⊕ax+1⊕⋯⊕ay=k的x,y有多少组。

 

输入

输入第一行为3个整数n,m,k。第二行为空格分开的n个整数，即a1,a2,…,an。接下来m行，每行两个整数lj,rj，代表一次查询。

 

输出

输出共m行，对应每个查询的计算结果。

 

样例输入

4 5 1
1 2 3 1
1 4
1 3
2 3
2 4
4 4

 

样例输出

4
2
1
2
1

 

提示

对于30%的数据，1≤n,m≤1000。
对于100%的数据，1≤n,m≤105,0≤k,ai≤105,1≤lj≤rj≤n。

思路：莫队算法

让a[i]为前i个位置的异或和，如果i—j的异或和等于k，那么a[i-1]^a[j]=k，a[i-1]^k=a[j]；所以莫队会把每个区间的a都会访问，因为是一个一个的移动，所以所有的都会读一遍，用一个cnt数组保存a[j]，所以每次移动一次就加或减一次cnt[a[i-1]^k];

初始化cnt[0] = 1：如果只给你一个数据ai = 1，和k = 1，怎么计算？就是1^0 = 1，就是说使用了数字0来参与异或，所以初始化cnt[0] = 1；同时，因为多使用了数字0，导致使用莫队模板时左端点 L-1

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n,m,k;
int a[100010],pos[100010],cnt[100010];
int res[100010];
int ans;
struct node
{
    int l,r,index;
}p[100010];
bool cmp(node a,node b)
{
    if(pos[a.l]==pos[b.l])
        return a.r<b.r;
    return a.l<b.l;
}
void add(int index)
{
    ans+=cnt[a[index]^k];
    cnt[a[index]]++;
}
void del(int index)
{
    ans-=cnt[a[index]^k];
    cnt[a[index]]--;
}
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    int i;
    for(i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        a[i]^=a[i-1];
    }
    int limt=sqrt(n);
    for(i=1;i<=n;i++)
    {
        pos[i]=(i-1)/limt+1;
    }
    for(i=1;i<=m;i++)
    {
        scanf("%d%d",&p[i].l,&p[i].r);
        p[i].index=i;
    }
    sort(p+1,p+1+m,cmp);

    int tl=1,tr=0;
    cnt[0]=1;
    ans=0;
    for(i=1;i<=m;i++)
    {
        while(tl<p[i].l)
        {
            del(tl-1);
            tl++;
        }
        while(tl>p[i].l)
        {
            tl--;
            add(tl-1);
        }
        while(tr<p[i].r)
        {
            add(++tr);
        }
        while(tr>p[i].r)
        {
            del(tr--);
        }
        res[p[i].index]=ans;
    }
    for(i=1;i<=m;i++)
        printf("%d\n",res[i]);


    return 0;
}