当前位置: 代码迷 >> 综合 >> 【SDOI2009】bzoj1878 HH的项链【解法二】
  详细解决方案

【SDOI2009】bzoj1878 HH的项链【解法二】

热度:78   发布时间:2024-01-13 11:28:39.0

Description HH有一串由各种漂亮的贝壳组成的项链。HH相信不同的贝壳会带来好运,所以每次散步
完后,他都会随意取出一段贝壳,思考它们所表达的含义。HH不断地收集新的贝壳,因此,
他的项链变得越来越长。有一天,他突然提出了一个问题:某一段贝壳中,包含了多少种不同
的贝壳?这个问题很难回答。。。因为项链实在是太长了。于是,他只好求助睿智的你,来解 决这个问题。 Input
第一行:一个整数N,表示项链的长度。 第二行:N个整数,表示依次表示项链中贝壳的编号(编号为0到1000000之间的整数)。
第三行:一个整数M,表示HH询问的个数。 接下来M行:每行两个整数,L和R(1 ≤ L ≤ R ≤ N),表示询问的区间。 Output
M行,每行一个整数,依次表示询问对应的答案。

解法一【离线+树状数组】见【这里】
莫队算法,时间复杂度O(n * sqrt(m)+m * sqrt(n))。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
struct qry
{int l,r,num;
}q[200010];
bool cmp_l(qry q1,qry q2)
{return q1.l<q2.l;
}
bool cmp_r(qry q1,qry q2)
{return q1.r<q2.r;
}
int n,m,a[50010],cnt[1000010],ans[200010],T,L[1010],R[1010];
int main()
{int i,j,k,x,y,z,now=0,tot;scanf("%d",&n);for (i=1;i<=n;i++)scanf("%d",&a[i]);scanf("%d",&m);for (i=1;i<=m;i++)scanf("%d%d",&q[i].l,&q[i].r),q[i].num=i;sort(q+1,q+m+1,cmp_l);T=sqrt(n);for (i=1;R[i-1]<m;i++){R[i]=L[i]=R[i-1]+1;while (R[i]<m&&q[R[i]+1].l-q[L[i]].l+1<=T) R[i]++;}tot=i-1;for (i=1;i<=tot;i++)sort(q+L[i],q+R[i]+1,cmp_r);for (i=1;i<=m;i++){if (q[i].l>q[i-1].l)for (j=q[i-1].l;j<q[i].l;j++){cnt[a[j]]--;if (cnt[a[j]]==0) now--;}elsefor (j=q[i].l;j<q[i-1].l;j++){cnt[a[j]]++;if (cnt[a[j]]==1) now++;}if (q[i].r<q[i-1].r)for (j=q[i].r+1;j<=q[i-1].r;j++){cnt[a[j]]--;if (cnt[a[j]]==0) now--;}elsefor (j=q[i-1].r+1;j<=q[i].r;j++){cnt[a[j]]++;if (cnt[a[j]]==1) now++;}ans[q[i].num]=now;}for (i=1;i<=m;i++)printf("%d\n",ans[i]);
}