Jzoj3901 二叉查找树

227 篇文章 3 订阅
134 篇文章 0 订阅

题意:给你一个bst插入的顺序问你完成这些操作需要多少次比较(即这颗bst所有节点深度之和)

显然,我们在bst中插入一个节点x,它一定是它前继或者后续的儿子

这样的话,我们记录每个点的深度,每次找前继和后续即可

本来应该用链表的,我还是为了卡时用了ZKW线段树

#pragma GCC opitmize("O3")
#pragma G++ opitmize("O3")
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 300010
using namespace std;
int s[N<<2],d[N],n,M; long long A=0;
inline void add(int x){ for(x+=M;x;x>>=1) ++s[x]; }
inline int pre(int x){
	for(x+=M;x;x>>=1)
		if((x&1)&&(s[x^1])){
			for(--x;x<=M;x=(s[x<<1|1]?x<<1|1:x<<1));
			return x-M;
		}
	return 0;
}
inline int suc(int x){
	for(x+=M;x;x>>=1)
		if((~x&1)&&(s[x^1])){
			for(++x;x<=M;x=(s[x<<1]?x<<1:x<<1|1));
			return x-M;
		}
	return 0;
}
int main(){
	scanf("%d",&n);
	for(M=1;M<n;M<<=1); --M;
	for(int x,i=0;i<n;++i){
		scanf("%d",&x); add(x);
		d[x]=max(d[pre(x)],d[suc(x)])+1;
		printf("%lld\n",A+=d[x]-1);
	}
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值