P1972 [SDOI2009]HH的项链[离线+树状数组/主席树/分块/模拟]

题目背景

题目描述

HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运,所以每次散步完后,他都会随意取出一段贝壳,思考它们所表达的含义。HH 不断地收集新的贝壳,因此,他的项链变得越来越长。有一天,他突然提出了一个问题:某一段贝壳中,包含了多少种不同的贝壳?这个问题很难回答……因为项链实在是太长了。于是,他只好求助睿智的你,来解决这个问题。

输入输出格式

输入格式:

第一行:一个整数N,表示项链的长度。

第二行:N 个整数,表示依次表示项链中贝壳的编号(编号为0 到1000000 之间的整数)。

第三行:一个整数M,表示HH 询问的个数。

接下来M 行:每行两个整数,L 和R(1 ≤ L ≤ R ≤ N),表示询问的区间。

输出格式:

M 行,每行一个整数,依次表示询问对应的答案。

输入输出样例

输入样例#1:

6
1 2 3 4 3 5
3
1 2
3 5
2 6

输出样例#1:

2
2
4

说明

数据范围:

对于100%的数据,N <= 500000,M <= 500000。

解析:

我居然为了这道题去学了主席树???结果发现树状数组秒解???主席树还MLE???

我真是超级蒻啊,居然真不会写。。。

口胡请见谅。。。


正解:离线+树状数组/主席树/分块/模拟(雾

我们分析题目,会发现实际上题目就是要我们求一个区间内的不同的数。

我们有一个巧妙的模板解法,可以用不同的数据结构实现:

以每个不同的位置建立数据结构。

我们设一个数组(a[i]=num)表示在(i)位置上,总共有(num)种不同的数。

对于修改:

对于任意的一个数列,我们按照顺序将这个数列从(1-n)的位置上的数依次加入(a)数组,我们会发现,实际上我们只需维护在当前插入位置(i)下,(1-i)中各个不同的数出现的最后的位置。

对于询问:

题解有一个(dalao)总结的精辟:对于若干个询问的区间[l,r],如果他们的r都相等的话,那么项链中出现的同一个数字,一定是只关心出现在最右边的那一个的

当然这种做法相当于告诉我们要边处理插入边记录答案,而不能处理完后再记录答案。

对于数组(a),数的位置是移动的。

举个例子:

(1~2~1~3~4)

这个数列中,我们的(a)数组原先是

(0~0~0~0~0)

(a)数组加入位置(1)(1)

(1~0~0~0~0)

加入位置(2)(2)

(1~1~0~0~0)

加入位置(3)(1)

(0~1~1~0~0)

就到这里为止,我们就能得出结论:对于一个区间内相同的数字,我们只用关心它最后出现的位置,这样相当于我们只允许所有相同的数字在这个区间内出现一次。

这样我们就可以得出在某一个区间不同的数的总数了,就是插入(i)位置的数时(a)数组中所有(1)的个数。

参考代码:

AC 树状数组

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<string>
#include<cstdlib>
#include<queue>
#include<vector>
#define INF 0x3f3f3f3f
#define PI acos(-1.0)
#define N 500010
#define MOD 2520
#define E 1e-12
using namespace std;
int ask[N<<1],n,m,last[N<<1],a[N<<1],ans[N<<1];
struct rec{
	int l,r;
	int id;
}q[N];
int query(int x)
{
	int ans=0;
	for(;x;x-=x&-x) ans+=ask[x];
	return ans;
}
void add(int x,int y)
{
	for(;x<=n;x+=x&-x) ask[x]+=y;
}
bool cmp(rec a,rec b)
{
	return a.r<b.r;
}
int main()
{
	scanf("%d",&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].id=i;
	sort(q+1,q+m+1,cmp);
	int next=1;
	for(int i=1;i<=m;i++){
		for(int j=next;j<=q[i].r;j++){
			int v=a[j];
			if(last[v])
				add(last[v],-1);
			add(j,1);
			last[v]=j;
		}
		next=q[i].r+1;
		ans[q[i].id]=query(q[i].r)-query(q[i].l-1);
	}
	for(int i=1;i<=m;i++) printf("%d
",ans[i]);
	return 0;
}

无耻的贴上MLE2个点的主席树:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<string>
#include<cstdlib>
#include<queue>
#include<vector>
#define INF 0x3f3f3f3f
#define PI acos(-1.0)
#define N 500010
#define MOD 2520
#define E 1e-12
#define ri register int
using namespace std;
struct tree{
	int lc,rc;
	int sum;
}t[N<<6];
int tot,n,m,root[N*20],last[N*20];
inline int read()
{
	int f=1,x=0;char c=getchar();
	while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
	return x*f;
}
inline int build(int l,int r)
{
	int p=++tot;
	if(l==r) return p;
	int mid=(l+r)>>1;
	t[p].lc=build(l,mid);
	t[p].rc=build(mid+1,r);
	return p;
}
inline int change(int pre,int v,int l,int r,int val)
{
	int p=++tot;
	t[p].lc=t[pre].lc,t[p].rc=t[pre].rc,t[p].sum=t[pre].sum+v;
	if(l==r) return p;
	int mid=(l+r)>>1;
	if(val<=mid) t[p].lc=change(t[pre].lc,v,l,mid,val);
	else t[p].rc=change(t[pre].rc,v,mid+1,r,val);
	return p;
}
inline int query(int x,int pre,int l,int r)
{
	if(l==r) return t[pre].sum;
	int mid=(l+r)>>1;
	if(x<=mid) return query(x,t[pre].lc,l,mid)+t[t[pre].rc].sum;
	else return query(x,t[pre].rc,mid+1,r);
}
int main()
{
	n=read();
	root[0]=t[0].lc=t[0].rc=t[0].sum;
	for(ri i=1;i<=n;i++){
		int x;
		x=read();
		if(last[x]){
			int t=change(root[i-1],-1,1,n,last[x]);
			root[i]=change(t,1,1,n,i);
		}
		else{
			root[i]=change(root[i-1],1,1,n,i);
		}
		last[x]=i;
	}
	
	m=read();
	for(ri i=1;i<=m;i++)
	{
		int l,r;
		l=read(),r=read();
		printf("%d
",query(l,root[r],1,n));
	}
	return 0;
}
原文地址:https://www.cnblogs.com/DarkValkyrie/p/11165936.html