CodeForces 617E XOR and Favorite Number 【莫队】

来源:互联网 发布:汉高祖知乎 编辑:程序博客网 时间:2024/05/20 07:34

题目:点击打开链接

题意:给定一个数列,求询问区间内任意区间段的连续异或值等于k的个数。

思路:对于异或运算,al,a(l+1)...ar的异或值等于a1,a2...a(l-1)的异或值异或上a1,a2...ar的值,故我们可以直接保存前缀异或和,用莫队算法,cnt数组表示在区间内出现的个数,由于x^k==y,x^y==k,则插入一个数时,答案加上该数异或k的cnt值,该数的cnt值加一,删除一个数时,先该数的cnt值减一,再答案减去该数异或k的cnt值,避免k异或自身

代码:

#include <stdio.h>#include <cmath>#include <algorithm>#include <queue>#include <set>#include <vector>#include <map>#define PR pair<int,int>#define MP make_pair#define fi first#define se second#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1#define sqr(x) ((x)*(x))#define LL long longconst LL INF = 1e18;const int inf=0x3f3f3f3f;const int M=18;const int N=1<<20;const int MOD=100000;const double eps=1e-10;const double pi=acos(-1.0);using namespace std;int n,m,k,sz,L,R,a[N],pos[N];LL cnt[N],Ans,ans[N];struct node{    int l,r,id;}Q[N];bool cmp(node a,node b){    if(pos[a.l]==pos[b.l]) return a.r<b.r;    return pos[a.l]<pos[b.l];}void add(int x){    Ans+=cnt[a[x]^k];    cnt[a[x]]++;}void del(int x){    cnt[a[x]]--;    Ans-=cnt[a[x]^k];}int main(){    int i;    scanf("%d%d%d",&n,&m,&k);    sz=sqrt(n);    for(i=1;i<=n;i++)scanf("%d",&a[i]),a[i]=a[i]^a[i-1],pos[i]=i/sz;    for(i=1;i<=m;i++)scanf("%d%d",&Q[i].l,&Q[i].r),Q[i].id=i;    sort(Q+1,Q+1+m,cmp);    cnt[0]=1,L=1,R=0,Ans=0;    for(i=1;i<=m;i++)    {        while(L<Q[i].l) del(L-1),L++;        while(L>Q[i].l) L--,add(L-1);        while(R<Q[i].r) R++,add(R);        while(R>Q[i].r) del(R),R--;        ans[Q[i].id]=Ans;    }    for(i=1;i<=m;i++)printf("%I64d\n",ans[i]);}


0 0
原创粉丝点击