深度剖析主席树(静态)

主席树(chairman tree)(据说是被chairman传开的)

它是一种专门用来实现求区间第K大这类操作的数据结构

 

Q:这不是分块轻松解决的事吗?

A:是的,但是分块时间为n\sqrt{n}的,而主席树是n lg n

Q:这是不是很难,代码复杂度极高?

A:NoNoNo,代码复杂度一点都不高,前置知识只有动态开点线段树

Q:那我不会怎么办?

A:右转自行度娘,或者菜鸡的博客  

 

菜鸡很菜,刚学了主席树,还是最简单的静态

首先思考一下,区间求第K大,最暴力的解自然是每次排序,是O(n^{2} lg n)

Q:如果用数据结构维护呢?

 A:每次询问用一棵权值线段树或者权值树状数组+二分维护,也是O(n^{2} lg n),QAQ,(不会可见可怜的狗狗这道题)

 

权值树状数组+二分姑且不论,我们只谈权值线段树(时间复杂度低才是王道)

每次询问的是一段区间,与区间相关的算法没有多少,也就前缀和,线段树……

权值线段树下标记的是值的大小,存的是出现次数,所以具有可加减性

故选用前缀和

——————————————————————————————————————

前面的引导就差不多结束了,进入正文了

 

主席树=前缀和+线段树=对每个前缀和中的点开一棵权值线段树(下标记的是权值的线段树)

线段树上的结点记录[1,i]个数中,大小在区间[l,r]中的数的个数

 

建立n棵线段树,不动态开点必定超空间,动态开点后,将新的树建立在原来的树之上

每次只用新插入一个结点,

如果该结点应该在左子树,右子树就不用变;反之左子树不用变

深度剖析主席树(静态)_第1张图片

每次新插入只用lg n,总共建n lg n个结点就够了

	void add(int &p,int l,int r,int x)
	{
		lc[++cnt]=lc[p],rc[cnt]=rc[p],s[cnt]=s[p]+1,p=cnt;
		if(l==r) return;
		int mid=l+r>>1;
		if(x<=mid) add(lc[p],l,mid,x);
			else add(rc[p],mid+1,r,x);
	}  //lc为左儿子,rc为右儿子,s为结点个数

 

查询:由于上文提到线段树具有可减性,只用在[1,R]的线段树-[1,L-1]的线段树找第K大,这算是权值线段树的基本操作吧

(不会可见可怜的狗狗这道题,也可以看代码)

	int kth(int p1,int p2,int l,int r,int x)
	{
		if(l==r) return l;
		int xx=s[lc[p2]]-s[lc[p1]];
		int mid=l+r>>1;
		if(x<=xx) return kth(lc[p1],lc[p2],l,mid,x);
			else return kth(rc[p1],rc[p2],mid+1,r,x-xx); 
	}

 

总结:主席树就是前缀和+线段树的思想毒瘤了一点(有心人可以看出我上面的引导是倒推的),但是只要你熟练掌握线段树,知道思想后小菜一碟

例题(时刻更新)

1、Luogu P3834 【模板】可持久化线段树 1(主席树)

   额,模板题,加个离散化就号了

   

#include
#include
using namespace std;

const int N=4e6+5;
int n,m,rt[N],b[N];
struct A{int id,x; }a[N];

bool cmp(A x,A y)
{
	return x.x>1;
		if(x<=mid) add(lc[p],l,mid,x);
			else add(rc[p],mid+1,r,x);
	}
	
	int kth(int p1,int p2,int l,int r,int x)
	{
		if(l==r) return l;
		int xx=s[lc[p2]]-s[lc[p1]];
		int mid=l+r>>1;
		if(x<=xx) return kth(lc[p1],lc[p2],l,mid,x);
			else return kth(rc[p1],rc[p2],mid+1,r,x-xx); 
	}
}zxs;
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) 
		a[i].id=i,scanf("%d",&a[i].x);
	sort(a+1,a+n+1,cmp);
	int tot=0;
	for(int i=1;i<=n;i++) b[a[i].id]=i;
	for(int i=1;i<=n;i++)
		rt[i]=rt[i-1],zxs.add(rt[i],1,n,b[i]);
	while(m--)
	{
		int l,r,k; scanf("%d%d%d",&l,&r,&k);
		printf("%d\n",a[zxs.kth(rt[l-1],rt[r],1,n,k)].x);
	}
	return 0;
}

2、可怜的狗狗(这是超链接)

模板题*2

3、Luogu P3919 【模板】可持久化数组(可持久化线段树/平衡树)

模板题*3

4、BZOJ 2223: [Coci 2009]PATULJCI 

模板题,查询中位数相同的是否超过序列长度的一半

你可能感兴趣的:(主席树,深度剖析系列)