莫队算法

在这里插入图片描述
洛谷P2709为例来一个莫队算法的模板:
在这里插入图片描述

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define INF 0x3f3f3f3f
const ll maxn=5e4+5;
ll a[maxn],cnt[maxn],ans[maxn];
struct node{
	ll l,r,id;
}q[maxn];
ll n,m,k,sz,sum;
bool cmp(node a,node b){
	if((a.l/sz)!=(b.l/sz))
		return a.l<b.l;
	return a.r<b.r;		
}
void add(int x){
    if(x<1||x>k) return;
    sum+=2*cnt[x]+1;
    cnt[x]++; 
}

void del(int x){
    if(x<1||x>k) return;
    sum-=2*cnt[x]-1;
    cnt[x]--;
}
int main(){
	cin>>n>>m>>k;
	sz=sqrt(n);
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	for(int i=1;i<=m;i++){
		cin>>q[i].l>>q[i].r;
		q[i].id=i;
	}
	sort(q+1,q+m+1,cmp);
	ll L=0,R=0;
	for(int i=1;i<=m;i++){
		while(R<q[i].r) add(a[R+1]),R++;
		while(R>q[i].r) del(a[R]),R--;
		while(L<q[i].l) del(a[L]),L++;
		while(L>q[i].l) add(a[L-1]),L--;
		ans[q[i].id]=sum;
	}
    for(int i=1;i<=m;i++) printf("%lld\n",ans[i]);
    return 0;		
}

再来一道例题:https://codeforces.com/contest/617/problem/E
这里4个特判需要注意变化,因为用的是前缀和

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define INF 0x3f3f3f3f
const ll maxn=1<<20;
ll a[maxn],cnt[maxn],ans[maxn];
struct node{
	ll l,r,id;
}q[maxn];
ll n,m,k,sz,sum;
bool cmp(node a,node b){
	if((a.l/sz)!=(b.l/sz))
		return a.l<b.l;
	return a.r<b.r;		
}
void add(int x){
    sum+=cnt[x^k];
    cnt[x]++; 
}

void del(int x){
    cnt[x]--;
    sum-=cnt[x^k];    
}
int main(){
	cin>>n>>m>>k;
	sz=sqrt(n);
	for(int i=1;i<=n;i++){
		cin>>a[i];
		a[i]=a[i]^a[i-1];
	}
	for(int i=1;i<=m;i++){
		cin>>q[i].l>>q[i].r;
		q[i].id=i;
	}
	sort(q+1,q+m+1,cmp);
	ll L=1,R=0;
	cnt[0]=1;
	for(int i=1;i<=m;i++){
		while(R<q[i].r) R++,add(a[R]);
		while(R>q[i].r) del(a[R]),R--;
		while(L<q[i].l) del(a[L-1]),L++;
		while(L>q[i].l) L--,add(a[L-1]);
		ans[q[i].id]=sum;
	}
    for(int i=1;i<=m;i++) printf("%lld\n",ans[i]);
    return 0;		
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值