BZOJ 3339 && luogu4137 Rmq Problem / mex(莫队)

P4137 Rmq Problem / mex

题目描述

有一个长度为n的数组{a1,a2,…,an}。m次询问，每次询问一个区间内最小没有出现过的自然数。

输入输出格式

输入格式：

第一行n,m。

第二行为n个数。

从第三行开始，每行一个询问l,r。

输出格式：

一行一个数，表示每个询问的答案。

输入输出样例

输入样例#1：

5 5

2 1 0 2 1

3 3

2 3

2 4

1 2

3 5

输出样例#1：

1

2

3

0

3

说明

对于30%的数据：1<=n,m<=1000

对于100%的数据：1<=n,m<=200000，0<=ai<=10^9，1<=l<=r<=n

简单莫队

不过这个转换不一定是O（n）的

注意细节

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <cstdio>
#define R register
#define maxn 301000
using namespace std;
int n,m,now;
int a[maxn];
int belong[maxn];
int vis[maxn];
int ma;
struct node {
	int l,r,id;
} q[maxn];
int ans[maxn];
inline int read()
{
	int x=0,f=1;char s=getchar();
	while('0'>s||s>'9') {
		if(s=='-') f=-1;
		s=getchar();
	}
	while('0'<=s&&s<='9') {
		x=x*10+s-'0';
		s=getchar();
	}
	return x*f;
}
inline bool cmp(node a,node b) {
	return belong[a.l]==belong[b.l] ? a.r<b.r : belong[a.l]<belong[b.l];
}
inline void add(int x) {
	++vis[x];
	if(now < x) return;
	if(now==x&&vis[x]==1) { //细节在这里
		for(R int i=now+1; i<n; ++i) {
			if(!vis[i])	{
				now=i;
				return;
			}
		}
	}
}
inline void delet(int x) {
	--vis[x];
	if(now < x) return;
	if(now > x) {
		if(!vis[x]) {
			now=x;
		}
	}
}
int main() {
	n=read();
	m=read();
	int k=sqrt(n);
	//别看a[]很大，ans范围小于n
	for(R int i=1; i<=n; ++i) {
		a[i]=read();
		if(a[i] > n)
			a[i]=n+1;
	}
	if(n<=1000&&m<=1000) { //baoli
		while(m--) {
			int l,r;
			memset(vis,0,sizeof(vis));
			l=read();
			r=read();
			for(R int i=l; i<=r; ++i)
				vis[a[i]]=1;
			R int i=0;
			while(vis[i]) i++;
			printf("%d\n",i);
		}
		return 0;
	}
	for(R int i=1; i<=n; ++i)
		belong[i]=(i-1)/k+1;
	for(R int i=1; i<=m; ++i)
	{
		q[i].l=read();
		q[i].r=read();
		q[i].id=i;
	}
	sort(q+1,q+1+m,cmp);
	int l=1,r=0;
	for(R int i=1; i<=m; ++i) {
		while(l > q[i].l) add(a[--l]);
		while(l < q[i].l) delet(a[l++]);
		while(r < q[i].r) add(a[++r]);
		while(r > q[i].r) delet(a[r--]);
		ans[q[i].id]=now;
	}
	for(R int i=1; i<=m; ++i)
		printf("%d\n",ans[i]);
	return 0;
}