莫队算法
因为在牛客国庆欢乐赛第一场中的签到题而现学了莫队算法
链接:https://ac.nowcoder.com/acm/contest/20322/J
来源:牛客网
Given a sequence of integers a1, a2, …, an and q pairs of integers (l1, r1), (l2, r2), …, (lq, rq), find count(l1, r1), count(l2, r2), …, count(lq, rq) where count(i, j) is the number of different integers among a1, a2, …, ai, aj, aj + 1, …, an.
数据范围 0<n,q<1e5
莫队签到就离谱
一开始想的朴素算法是在线处理每个询问
假设数组元素全部相同,遍历左边和右边,若没出现过则将答案加一
时间复杂度是恐怖的O(nq)。
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 +10;
bool flag[N];
int a[N];
int n,q;
int main(){
while(~scanf("%d %d",&n,&q)){
memset(a,0,sizeof a);
//memset(b,0,sizeof b);
//scanf("%d",&a1);
for(int i = 0;i < n;i ++ ){
cin>>a[i];
}
while(q -- ){
int cnt = 1;
int l,r;
memset(flag,false,sizeof flag);
flag[a[0]] = true;
scanf("%d %d",&l,&r);
for(int i = 1;i < l;i ++ ){
if(flag[a[i]] == false){
cnt += 1;
flag[a[i]] = true;
}
}
for(int i = r - 1;i < n;i ++ ){
if(flag[a[i]] == false){
cnt += 1;
flag[i] = true;
}
}
printf("%d\n",cnt);
}
}
}
显而易见的超时了
问题来了,超时显然是因为我们在线处理每个询问,那么,如何离线处理呢?
如果我们能够将查询分块,对于每次查询,更新块中的值
起初令左指针=0,右指针=0,如果当前查询的左端在左指针前,就将左指针从得到的集合中删除,在左指针后,就将左指针加入得到的集合中,右指针同理。
for(int i = 1,l = 1,r = 2;i <= q;i ++ ){
if(ask[i].l == ask[i].r){
ask[i].ans = temp;
continue;
}
while(r < ask[i].r) del[a[r ++ ]];
while(r > ask[i].r) add(a[-- r]);
while(l < ask[i].l) add(a[++ l]);
while(l > ask[i].l) del(a[l -- ]);//若在左指针左边,删除这个点
ask[i].ans = sum;
}
莫队算法就这样写出来了
贴上AC代码
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
struct node{
int l,r,ans,id;
}ask[N];
int n,q,sum,block;
int a[N],cnt[N],pos[N];
bool cmp(node x,node y){
return x.l / block == y.l / block ? x.l/block < y.l/block : x.r < y.r;
}
bool CMP(node x,node y){
return x.id < y.id;
}
void add(int x){
cnt[x] ++;
if(cnt[x] == 1) sum ++;
}
void del(int x){
if(cnt[x] == 1) sum --;
cnt[x] --;
}
int main(){
int n,q;
while(~scanf("%d %d",&n,&q)){
sum = 0;
block = sqrt(1 * n);
for(int i = 1;i <= n;i ++ ){
scanf("%d",&a[i]);
if(!cnt[a[i]]) sum ++;
cnt[a[i]] ++;
}
int temp = sum;
for(int i = 1;i <= q;i ++ ){
scanf("%d %d",&ask[i].l,&ask[i].r);
ask[i].id = i;
}
sort(ask + 1,ask + q + 1,cmp);
int l = 0,r = 0;
for(int i = 1,l = 1,r = 2;i <= q;i ++ ){
if(ask[i].l == ask[i].r){
ask[i].ans = temp;
continue;
}
while(r < ask[i].r) del[a[r ++ ]];
while(r > ask[i].r) add(a[-- r]);
while(l < ask[i].l) add(a[++ l]);
while(l > ask[i].l) del(a[l -- ]);
ask[i].ans = sum;
}
sort(ask + 1,ask + q + 1,CMP);
for(int i = 1;i <= q;i ++ )
printf("%d\n",ask[i].ans);
}
return 0;
}
时间复杂度O(n
n
\sqrt{n}
n)
在此orz一波莫涛队长,让暴力也变得如此优美