码迷,mamicode.com
首页 > 其他好文 > 详细

bzoj-2212 Tree Rotations

时间:2015-09-30 13:08:06      阅读:210      评论:0      收藏:0      [点我收藏+]

标签:bzoj   poi   线段树   线段树合并   

题意:

给出一个二叉树,叶子结点上有权值;

共有n个叶子结点,权值分别为1-n;

现可将所有结点的左右儿子交换,求最小的逆序对数;

2<=n<=200000;


题解:

这题读入有点鬼畜,但是写起来还是比较优雅的;

考虑到一个结点的子树具体形态与它是否和它的兄弟交换是无关的;

所以我们可以像分治一样,先计算子树之间的最小逆序对数;

然后将所有的权值扔到一个线段树中,向它的父亲传递;

它的父亲是没有权值的,我们只是在这里计算这两个线段树之间最小逆序对数,并将其合并就好了;

具体实现实际上只是一个合并操作中多了两个记录,考虑左子树与右子树是否交换;

对于一个线段树上的结点,我们累加:

左子树<=mid的结点数*右子树>mid的结点数作为左子树与右子树交换的逆序对数;

左子树>mid的结点数*右子树<=mid的结点数作为左子树与右子树不交换的逆序对数;

然后合并完成之后比较一下,累加到答案里就可以了;

答案可能超过int,时间复杂度O(nlogn)吧。。。

内存似乎比较玄学,应该是不超过nlogn的(空间复杂度不会大过时间复杂度呢);


代码:

#include<cctype>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 210000
#define M 3000000
#define LEN 1<<15
#define lson l,mid,ls[no]
#define rson mid+1,r,rs[no]
using namespace std;
typedef long long ll;
int sum[M],ls[M],rs[M],tot;
int st[M],top,n;
ll ans;
char getc()
{
	static char *S,*T,buf[LEN];
	if(S==T)
	{
		T=(S=buf)+fread(buf,1,LEN,stdin);
		if(S==T)
			return EOF;
	}
	return *S++;
}
int read()
{
	static char ch;
	static int D;
	while(!isdigit(ch=getc()));
	for(D=ch-'0';isdigit(ch=getc());)
		D=D*10+ch-'0';
	return D;
}
int newnode()
{
	if(top)	return st[top--];
	return ++tot;
}
void delnode(int no)
{
	st[++top]=no;
	sum[no]=ls[no]=rs[no]=0;
}
void Pushup(int no)
{
	sum[no]=sum[ls[no]]+sum[rs[no]];
}
void Insert(int l,int r,int &no,int val)
{
	if(!no)	no=newnode();
	if(l==r)
		sum[no]++;
	else
	{
		int mid=l+r>>1;
		if(val<=mid)Insert(lson,val);
		else		Insert(rson,val);
		Pushup(no);	
	}
}
int merge(int l,int r,int nol,int nor,ll &cntl,ll &cntr)
{
	if(!nol||!nor)	return nol+nor;
	cntl+=(ll)sum[rs[nol]]*sum[ls[nor]];
	cntr+=(ll)sum[ls[nol]]*sum[rs[nor]];
	int mid=l+r>>1,no=newnode();
	ls[no]=merge(l,mid,ls[nol],ls[nor],cntl,cntr);
	rs[no]=merge(mid+1,r,rs[nol],rs[nor],cntl,cntr);
	Pushup(no);
	delnode(nol),delnode(nor);
	return no;
}
int dfs()
{
	int x;
	x=read();
	if(!x)
	{
		int ls=dfs();
		int rs=dfs();
		if(sum[ls]>sum[rs])
			swap(ls,rs);
		ll cntl=0,cntr=0;
		int rt=merge(1,n,ls,rs,cntl,cntr);
		ans+=min(cntl,cntr);
		return rt;
	}
	else
	{
		int rt=0;
		Insert(1,n,rt,x);
		return rt;
	}
}
int main()
{
	n=read();
	dfs();
	printf("%lld\n",ans);
	return 0;
}



bzoj-2212 Tree Rotations

标签:bzoj   poi   线段树   线段树合并   

原文地址:http://blog.csdn.net/ww140142/article/details/48826481

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!