[洛谷1972][SDOI2009]HH的项链
题目描述
HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运,所以每次散步完后,他都会随意取出一段贝壳,思考它们所表达的含义。HH 不断地收集新的贝壳,因此,他的项链变得越来越长。有一天,他突然提出了一个问题:某一段贝壳中,包含了多少种不同的贝壳?这个问题很难回答……因为项链实在是太长了。于是,他只好求助睿智的你,来解决这个问题。
输入输出格式
输入格式:
第一行:一个整数N,表示项链的长度。
第二行:N 个整数,表示依次表示项链中贝壳的编号(编号为0 到1000000 之间的整数)。
第三行:一个整数M,表示HH 询问的个数。
接下来M 行:每行两个整数,L 和R(1 ≤ L ≤ R ≤ N),表示询问的区间。
输出格式:
M 行,每行一个整数,依次表示询问对应的答案。
输入输出样例
说明
数据范围:
对于100%的数据,N <= 50000,M <= 200000。
莫队
用cnt[i]表示数i的出现次数,sum表示当前区间的数的种类。
移动指针时,记i为当前新增(删除)的数,对于使区间增大的移动(l--,r++)若cnt[i]==1(即意味着出现不同种类的数)(这里的cnt指更新后的cnt),sum++
否则cnt[i]==0(意味着有一种数没了),sum--
cnt表示的是数的出现次数,不是数的下标的出现次数,容易误打
#include<iostream> #include<cstdio> #include<algorithm> #include<cmath> using namespace std; struct xxx{ int l,r,block,id; }q[200001]; int a[50001],cnt[1000100],ans[200001]; bool cmp(xxx a,xxx b){if(a.block!=b.block)return a.block<b.block;return a.r<b.r;} int main() { int n,m;scanf("%d",&n);int T=(int)sqrt((double)n); for(int i=1;i<=n;i++)scanf("%d",&a[i]); scanf("%d",&m); for(int i=1;i<=m;i++) { scanf("%d%d",&q[i].l,&q[i].r);q[i].block=(q[i].l+1)/T;q[i].id=i; } sort(q+1,q+m+1,cmp); int l=1,r=0,sum=0; for(int i=1;i<=m;i++) { while(r<q[i].r){cnt[a[++r]]++;if(cnt[a[r]]==1)sum++;} while(r>q[i].r){cnt[a[r]]--;if(cnt[a[r--]]==0)sum--;} while(l<q[i].l){cnt[a[l]]--;if(cnt[a[l++]]==0)sum--;} while(l>q[i].l){cnt[a[--l]]++;if(cnt[a[l]]==1)sum++;} ans[q[i].id]=sum; } for(int i=1;i<=m;i++)printf("%d\n",ans[i]); return 0; }