您的位置:首页 > 其它

莫队算法模板

2017-10-23 20:33 246 查看
codeforces 617E

题意:给你n个数,有m个询问,问[l,r]之间有多少对i和j满足a[i]^a[i+1]^...^a[j]=k;

这个题目就是我先求出前缀异或和

然后i到j的异或和就是a[j]^a[i-1]

假设我们已知st到en的答案(ans),然后我们开一个Map数组,Map[i]表示st到en中前缀异或和为i的个数,Map也是已知的

现在我们要加上一个点,st-1或者en+1,假设这个点的位置是poi

那么ans+=Map[a[poi]^k],Map[a[poi]]++;即可

同理我们也可以知道如何减去一个数

然后用莫队算法就行了

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <iostream>
#define maxn 2001000
#define ll long long

using namespace std;

ll n,m,K,fz;
ll a[maxn];

struct Data
{
ll l,r,id;
};

Data que[maxn];

bool cmp(Data aa,Data bb)//按莫队算法排序
{
if ((aa.l/fz)==(bb.l/fz))
return aa.r<bb.r;
else
return (aa.l/fz)<(bb.l/fz);
}

ll ans(0),Map[maxn],Ans[maxn];

void add(ll poi)
{
ans+=Map[a[poi]^K];
Map[a[poi]]++;
}

void del(ll poi)
{
Map[a[poi]]--;
ans-=Map[a[poi]^K];
}

int main()
{
while(~scanf("%lld %lld %lld",&n,&m,&K))
{
a[0]=0;
for (ll k=1;k<=n;k++)
{
scanf("%lld",&a[k]);
a[k]=a[k]^a[k-1];
}

for (ll k=1;k<=m;k++)
{
scanf("%lld %lld",&que[k].l,&que[k].r);
que[k].id=k;
}

fz=sqrt(n);

sort(que+1,que+m+1,cmp);

ll l(0),r(1);
memset(Map,0,sizeof(Map));
Map[a[1]]++;
Map[a[0]]++;
ans=(a[1]==K);

for (ll k=1;k<=m;k++)//进行处理
{
while(l<que[k].l-1)//这边要不要-1要具体问题具体分析
{
del(l);//是先++还是del,还是反过来也要具体分析
l++;
}

while(l>que[k].l-1)
{
l--;
add(l);
}

while(r>que[k].r)
{
del(r);
r--;
}

while(r<que[k].r)
{
r++;
add(r);
}

Ans[que[k].id]=ans;
}//完结撒花

for (ll k=1;k<=m;k++)
printf("%lld\n",Ans[k]);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: