[POI2008]砖块Klo

那么问题就在于如何找任意区间的中位数

划分树和平衡树都可以解决

所以是水题一道……

下面描述我的悲惨经历……

上周周末就看到了这道题

觉得应该可以用划分树水过

屁颠屁颠写了个划分树,写到一半发现不对头,好像不能做

和pty讨论了一下发现实际上还是可以做

只不过维护很蛋疼……

然后因为要期末考试了,把它放在那里没管

今天有空打算写一下

然后想的时候又觉得划分树没法做,然后就被绕进去了……

果断把代码一把删掉……

结果实际上划分树是真的可以做……

然后想一想还是写平衡树好了

然后再想一想发现multiset可以水过

虽然STL不支持找第k大 //也许是出于对速度的保证,STL并没有维护sum,size等“多余”的信息

但是因为新加进来一个数中位数只会向前向后挪一位,因此还是可求的

写的时候发现各种菊紧,比如说删掉了中位数之类的,STL要各种特判

然后火了一把删掉,手写Splay

写Splay的时候心血来潮想把它写成class的形式

许久没写Splay结果zigzag差点写疵

以前一般都写的静态Splay,这里来个动态的让我有点小纠结……

然后写啊写啊写……

写完了,wa啊wa啊wa……

最后发现Del过程写疵了……

首先是删点的时候size没有正确维护……

还有就是如果只有一个点的话,不能正确删除……//我没有判断把要删除的点旋到根以后左右子树都为空的情况……

结果k=1的点统统tle……

以上……


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

#include<iostream>
#include<algorithm>
#include<vector>
#include<string>
#include<queue>
#include<set>
#include<map>
using namespace std;
//Macro
#define	rep(i,a,b)	for(int i=a,tt=b;i<=tt;++i)
#define	drep(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=1e100;
const double eps=1e-6;
string name="", in=".in", out=".out";
//Var
struct T
{
	int LC,RC,FA,SIZE;
	ll SUM,KEY;
	#define lc(x) tree[x].LC
	#define rc(x) tree[x].RC
	#define fa(x) tree[x].FA
	#define size(x) tree[x].SIZE
	#define sum(x) tree[x].SUM
	#define key(x) tree[x].KEY
}tree[100008];
int tot,root;
int n,k;
ll ans,num[100008];
void Update(int x)
{
	sum(x)=sum(lc(x))+sum(rc(x))+key(x);
	size(x)=size(lc(x))+size(rc(x))+1;
}
void Zig(int x)
{
	int y=fa(x);
	if(lc(fa(y))==y)lc(fa(y))=x;else rc(fa(y))=x;fa(x)=fa(y);
	fa(rc(x))=y;lc(y)=rc(x);rc(x)=y;fa(y)=x;
	Update(y);
}
void Zag(int x)
{
	int y=fa(x);
	if(lc(fa(y))==y)lc(fa(y))=x;else rc(fa(y))=x;fa(x)=fa(y);
	fa(lc(x))=y;rc(y)=lc(x);lc(x)=y;fa(y)=x;
	Update(y);		
}
void Splay(int x)
{
	for(int y,z;fa(x);)
	{
		y=fa(x);z=fa(y);
		if(!z)
			if(lc(y)==x)Zig(x);
			else Zag(x);
		else
			if(lc(z)==y)
				if(lc(y)==x)Zig(y),Zig(x);
				else Zag(x),Zig(x);
			else 
				if(rc(y)==x)Zag(y),Zag(x);
				else Zig(x),Zag(x);
	}
	root=x;
	Update(x);
}
int Find(int k)
{
	int p=root;
	while(1)
	{
		if(size(lc(p))+1==k)return p;
		else if(size(lc(p))+1>k)p=lc(p);
		else k-=size(lc(p))+1,p=rc(p);
	}
}
void Ins(ll x)
{
	int f=0,p=root,flag=0;
	while(1)
	{
		if(p==0)
		{
			key(++tot)=x,fa(tot)=f;
			if(flag)rc(f)=tot;else lc(f)=tot;
			Splay(tot);
			break;
		}
		if(key(p)>=x)f=p,p=lc(p),flag=0;
		else f=p,p=rc(p),flag=1;
	}
}
void Del(int p)
{
	int q;
	Splay(p);
	if(!lc(p)&&!rc(p)){root=0;return;}
	if(lc(p)==0){fa(rc(p))=0;Splay(rc(p));}
	else if(rc(p)==0){fa(lc(p))=0;Splay(lc(p));}
	else
	{
		fa(lc(p))=0;
		for(q=lc(p);rc(q);q=rc(q));
		Splay(q);
		fa(rc(p))=q;rc(q)=rc(p);
		Update(q);
	}
}
ll Calc()
{
	int size=size(root)+1>>1;
	int p=Find(size);
	Splay(p);
	return key(p)*size(lc(p))-sum(lc(p))+sum(rc(p))-key(p)*size(rc(p));
}
void Init()
{
	ios::sync_with_stdio(0);
	cin>>n>>k;
	rep(i,1,n)cin>>num[i];
}
void Work()
{
	rep(i,1,k)
		Ins(num[i]);
	ans=Calc();
	rep(i,k+1,n)
	{
		Del(i-k);
		Ins(num[i]);
		ans=min(ans,Calc());
	}
	cout<<ans<<endl;
}
int main()
{
//	freopen((name+in).c_str(),"r",stdin);
//	freopen((name+out).c_str(),"w",stdout);
	Init();
	Work();
	return 0;
}

你可能感兴趣的:([POI2008]砖块Klo)