关于划分树

从昨天晚上一直到今天,我都在纠结划分树……TMD打错变量害死人……我还一直以为划分树哪里写错了……


划分树可以在O(nlogn)的时间内建树(常数巨大……),在logn的时间内求出一个区间内的第k小数(求第k大就改一下建树时的偏序关系)

怎么样做呢,HH大牛的blog上有一篇日志的图非常好……

此图片转载自http://www.notonlysuccess.com/index.php/divide-tree/

val[0]为初始序列,sorted为初始序列排序后的序列,sorted是为了方便的找到中位数

然后对于每一层的每个区间,找到这个区间的中位数,小于它分左边,大于它分右边,如图

然后对于每一层每个区间的每个点,记录这个点所在的区间到这个点为止有多少个数分到了左边


有一个问题,就是如果划分的时候碰到跟中位数相等的数怎么办呢

有两种办法,一种是先离散成1-n的排列,

还有一种办法是事先扫一遍这个区间,求得有多少个数与中位数相等,应该分多少个到左边去

第二种方法的常数会比第一种大,但是我一般不喜欢写离散(尽管就几行代码),使用哪种就看个人喜好了

/*

我曾经以为可以记一个same数组,这样可以快速得到这个区间里面有多少个数和中位数相等,加快速度

这种方法可以通过poj2104,但是实际上是错误的,在数据较大的情况下有较低概率出错,这证明poj2104数据太水

*/




而询问实际上就是一个不断逼近的过程,六元组(l,r,x,y,k,h)代表现在在树上第h层,l-r这个区间,询问x,y之间的第k小

由于建树时记录的信息,可以轻松求出x-y之间有多少个数被放到了下一层的左边,记为ss

如果ss>=k,那么去下一层的左子树找,否则去右子树找,根据这些信息也可以得到下一层的x,y(具体看代码)


可以发现query的过程只用到了build当中的  记录下一层有多少个在左边  的信息

所以划分树本身是可以开成滚动的



自认为自己的划分树还是写的比较好看的

核心代码30行不到(这里划分树没有开成滚动,滚动实测好像会慢一点)   poj2104:

//Lib
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<ctime>

#include<iostream>
#include<algorithm>
#include<vector>
#include<string>
#include<queue>
using namespace std;
//Macro
#define rep(i,a,b) for(int i=a,tt=b;i<=tt;++i)
#define rrep(i,a,b) for(int i=a,tt=b;i>=tt;--i)
#define erep(i,e,x) for(int i=x;i;i=e[i].next)
#define irep(i,x) for(__typedef(x.begin()) i=x.begin();i!=x.end();i++)
#define read() (strtol(ipos,&ipos,10))
#define sqr(x) ((x)*(x))
#define pb push_back
#define PS system("pause");
typedef long long ll;
typedef pair<int,int> pii;
const int oo=~0U>>1;
const double inf=1e20;
const double eps=1e-6;
string name="input",in=".in",out=".out";
//Var
int s[20][200008],d[20][200008],n,m,sorted[200008];
void Init();
void Work();
void Build(int l,int r,int h);
int Query(int l,int r,int x,int y,int k,int h);
int main()
{
	Init();
	Work();
	return 0;
}
void Init()
{
	scanf("%d%d",&n,&m);
	rep(i,1,n)scanf("%d",&d[1][i]),sorted[i]=d[1][i];
	sort(sorted+1,sorted+1+n);
	Build(1,n,1);
}
void Work()
{
	int x,y,k;
	rep(i,1,m)
	{
		scanf("%d%d%d",&x,&y,&k);
		printf("%d\n",Query(1,n,x,y,k,1));
	}
}
void Build(int l,int r,int h)//d数组为划分树,s数组为分多少到下一层左边的
{
	if(l==r)return;
	int mid=l+r>>1,cnt=mid-l+1,smid=sorted[mid],lpos=l,rpos=mid+1,pos=0;
	rep(i,l,r)if(d[h][i]<smid)cnt--;
	rep(i,l,r)
	{
		if(d[h][i]<smid||d[h][i]==smid&&cnt)
		{
			pos++;if(d[h][i]==smid)cnt--;
			d[h+1][lpos++]=d[h][i];
		}
		else d[h+1][rpos++]=d[h][i];
		s[h][i]=pos;
	}
	Build(l,mid,h+1);
	Build(mid+1,r,h+1);
}
int Query(int l,int r,int x,int y,int k,int h)
{
	if(l==r)return sorted[l];
	int mid=l+r>>1,l1,l2;
	l1=(l==x)?0:s[h][x-1];
	l2=s[h][y];
	if(k<=l2-l1)return Query(l,mid,l+l1,l+l2-1,k,h+1);
	else return Query(mid+1,r,mid+1+x-(l+l1),mid+1+y-(l+l2),k-(l2-l1),h+1);
}


你可能感兴趣的:(关于划分树)