CodeForces 617 E.XOR and Favorite Number（莫队算法）

最新推荐文章于 2020-05-21 20:54:01 发布

v5zsq

最新推荐文章于 2020-05-21 20:54:01 发布

阅读量272

点赞数

分类专栏： Code Forces 莫队算法

本文链接：https://blog.csdn.net/V5ZSQ/article/details/79859289

版权

Code Forces 同时被 2 个专栏收录

606 篇文章 6 订阅

订阅专栏

莫队算法

11 篇文章 0 订阅

订阅专栏

Description

给出 $n$ 个非负整数 $a_1,...,a_n$ 和 $m$ 组查询，每次查询给出一区间 $[l,r]$ ，查询满足 $l\le i\le j\le r$ 且 $a_i$ ^ $a_{i+1}$ ^ $...$ ^ $a_j=k$ 的 $(i,j)$ 个数

Input

第一行三个整数 $n,m,k$ ，之后输入 $n$ 个整数 $a_1,...,a_n$ ，最后 $m$ 行每行一组查询

$(1\le n,m\le 10^5,1\le k,a_i\le 10^6)$

Output

对于每组查询，输出查询结果

Sample Input

6 2 3
1 2 1 1 0 3
1 6
3 5

Sample Output

7
0

Solution

记 $s_i=a_1$ ^ $a_2$ ^ $...$ ^ $a_i$ ，则查询转化为求 $l\le i\le j\le r,s_{i-1}$ ^ $s_j=k$ 的个数，把所有查询左端点减一，则查询转化为求 $l \le i<j\le r,s_i$ ^ $s_j=k$ 的个数，固定右端点 $j$ ，则满足条件的左端点为 $i<j$ 且 $s_i$ = $s_j$ ^ $k$ ，固定左端点同理，那么只要记录 $[l,r]$ 中满足 $l\le i\le r,s_i=x$ 的 $i$ 个数 $num[x]$ ，那么当右端点从 $r$ 变成 $r+1$ 时，对答案的贡献为 $num[s_{r+1}$ ^ $k]$ ，右端点减一或者左端点加一减一同理，故每次区间端点移动一个位置只需要 $O(1)$ 的修改和更新贡献，用莫队算法 $O(n\sqrt{n})$ 即可求解

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 1111111
int n,m,k,a[maxn],sum[maxn],num[maxn],pos[maxn];
ll ans;
struct node
{
    int l,r,id;
    ll ans;
    bool operator<(const node &b)const
    {
        if(pos[l]!=pos[b.l])return l<b.l;
        return r<b.r;
    }
}q[maxn];
int cmp(node x,node y)
{
    return x.id<y.id;
}
void And(int x)
{
    x=sum[x];
    ans+=num[x^k];
    num[x]++;
}
void Plus(int x)
{
    x=sum[x];
    num[x]--;
    ans-=num[x^k];
}
int main()
{
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        memset(num,0,sizeof(num));
        int mm=(int)sqrt(n);
        sum[0]=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            sum[i]=sum[i-1]^a[i];
            pos[i]=(i-1)/mm+1;
        }
        for(int i=0;i<m;i++)
        {
            scanf("%d%d",&q[i].l,&q[i].r);
            q[i].id=i,q[i].l--;
        }
        sort(q,q+m);
        ans=0;
        num[0]=1;
        int l=0,r=0;
        for(int i=0;i<m;i++)
        {
            while(r<q[i].r)And(++r);
            while(r>q[i].r)Plus(r--);
            while(l<q[i].l)Plus(l++);
            while(l>q[i].l)And(--l);
            q[i].ans=ans; 
        }
        sort(q,q+m,cmp);
        for(int i=0;i<m;i++)printf("%I64d\n",q[i].ans);
    }
    return 0;
}