1878: [SDOI2009]HH的项链

看见题目第一眼就想到莫队了,为什么看网上一堆题解都是树状数组,果然我智商低吗QAQ。

算了,反正莫队乱搞就好了。

很明显,我们用一个数组记录编号为i的贝壳有多少个,增减的时候处理一下维护答案,即在O(1)的时间内从[L,R]到[L-1,R]或[L+1,R]或[L,R-1]或[L,R+1],然后就可以莫队了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=50000+5;
const int M=200000+5;
int have[1000010],pos[N],a[N],ans[M],now;
struct Query{
	int l,r,id;
	bool operator<(const Query& rhs)const{
		if(pos[l]!=pos[rhs.l])return pos[l]<pos[rhs.l];
		return pos[r]<pos[rhs.r];
	}
}q[M];
inline int read(){
	int x=0;char ch;
	while(ch<'0'||ch>'9')ch=getchar();
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x;
}
inline void add(int x){
	if(!have[x])now++;
	have[x]++;
}
inline void del(int x){
	have[x]--;
	if(!have[x])now--;
}
int main(){
	int n=read(),block=sqrt(n);
	for(int i=1;i<=n;i++){
		a[i]=read();
		pos[i]=(i-1)/block;
	}
	int m=read();
	for(int i=1;i<=m;i++){
		q[i].l=read();q[i].r=read();
		q[i].id=i;
	}
	sort(q+1,q+1+m);
	for(int i=q[1].l;i<=q[1].r;i++)
	add(a[i]);ans[q[1].id]=now;
	int ql=q[1].l,qr=q[1].r;
	for(int i=2;i<=m;i++){
		if(q[i].r<ql||qr<q[i].l){
			for(int j=ql;j<=qr;j++)del(a[j]);
			for(int j=q[i].l;j<=q[i].r;j++)add(a[j]);
		}else{
			if(ql<q[i].l)for(int j=ql;j<q[i].l;j++)del(a[j]);
			else for(int j=ql-1;j>=q[i].l;j--)add(a[j]);
			if(qr>q[i].r)for(int j=qr;j>q[i].r;j--)del(a[j]);
			else for(int j=qr+1;j<=q[i].r;j++)add(a[j]);
		}
		ans[q[i].id]=now;
		ql=q[i].l;qr=q[i].r;
	}
	for(int i=1;i<=m;i++)
	printf("%d\n",ans[i]);
	return 0;
}


你可能感兴趣的:(1878: [SDOI2009]HH的项链)