主席树(可持久化线段树)学习笔记

首先离散化 然后当前版本的线段树可以有上个版本的线段树修改lgn个节点得到的

还满足区间减法

好神


#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<set>
using namespace std;
set<int>D;
char c;
bool flag;
inline void read(int&a)
{a=0,flag=false;do c=getchar();while(c!='-'&&(c<'0'||c>'9'));if(c=='-')c=getchar(),flag=true;while(c<='9'&&c>='0')a=(a<<3)+(a<<1)+c-'0',c=getchar();if(flag)a=-a;}
struct Node
{
	Node *lc,*rc;
    int l,r,sum;
}*Segement_Root[10001];
struct O
{int x,z;}Line[100001];
int data[100001];
int Sorted_data[1000001];

Node*build(int l,int r)
{
	Node*tp=new Node;
	tp->lc=NULL;
	tp->rc=NULL;
	tp->l=l;
	tp->r=r;
	tp->sum=0;
   if(l^r)tp->lc=build(l,(l+r)>>1),tp->rc=build(((l+r)>>1)+1,r);	
   return tp;
}
Node *add(Node *last,int l,int r,int data)
{
	if(l^r)
	{
		Node *tp=new Node;
		*tp=*last;
		if(data<=Sorted_data[(l+r)>>1])
		    tp->lc=add(last->lc,l,(l+r)>>1,data);
		else tp->rc=add(last->rc,((l+r)>>1)+1,r,data);
	   tp->sum++;
	  return tp;
	}
    else
      {
      	Node *tp;
	    tp=new Node;
	    *tp=*last;
	    tp->sum++;
	    return tp;
	  }
}

int K_th(Node *rr,Node *lr,int l,int r,int k)
{
	if(l^r)
      {
      	if(rr->lc->sum-lr->lc->sum>=k)
      	     return K_th(rr->lc,lr->lc,l,(l+r)>>1,k);
      	return K_th(rr->rc,lr->rc,((l+r)>>1)+1,r,k-rr->lc->sum+lr->lc->sum);
	  }
	else return Sorted_data[l];
}
int main()
{
	int n,m;
	read(n);
	read(m);
	for(int i=1;i<=n;i++)
	   read(data[i]),D.insert(data[i]);
	set<int>::iterator tp;
	int con;
	for(tp=D.begin();tp!=D.end();tp++)
	    Sorted_data[++con]=*tp;
	Segement_Root[0]=build(1,con);
	for(int i=1;i<=n;i++) Segement_Root[i]=add(Segement_Root[i-1],1,con,data[i]);
	int l,r,k;
	while(m--)
	 {
	 	read(l),read(r),read(k);
	    printf("%d\n",K_th(Segement_Root[r],Segement_Root[l-1],1,con,k));
	 }
	return 0;
}


你可能感兴趣的:(主席树,可持久化线段树)