qbxt day7
t1. 树
【问题背景】
zhx 是二叉查找树大师。
【问题描述】
二叉查找树是一种特殊的二叉树(每个节点最多只有两个儿子的树)。树的
每个节点上存有一个唯一的值,并且满足:这个节点的左子树内所有点的值都
比这个节点的值小,且右子树内所有点的值都比这个节点的值要大。
对于一棵二叉查找树 T,我们可以将一个值为 x 的新点插入 T 中,且保持
树的性质。算法如下:
需要将 x 插入树 T 时,执行 insert(x, T.root) 。
现在有 N 个数需要插入一棵空树中。给定插入序列,请在每个元素被插入
之后,输出所有节点的深度总和(根的深度为 0)。
【输入格式】
输入的第一行一个整数 n,表示序列长度。
以下 n 行是序列中的数字,这些数字是各不相同的,在[1,n]区间。
【输出格式】
输出 n 行,第 i 行整数表示第 i 个数插入树后,至这个节点的节点深度总和。
【样例输入】
8
3
5
1
6
8
7
2
4
【样例输出】
0
1
2
4
7
11
13
15
【数据规模与约定】
对于 50%的数据,满足N≤ 1000
对于100%的数据,满足N≤ 3 ∗ 1e5。
纯模拟 50分
正解 建一个双向链表,当插入x点时,设有v < x < u, 其中u当前已插入的点中比x小的最大元, v是比x大的最小元。 因为要求顺序插入,则在处理时要倒过来找u,v; 然后正序模拟插入;
//自觉把读优删掉
#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=3*1e5+10;
int deep[maxn],u[maxn],v[maxn],prev[maxn],next[maxn],a[maxn],n,x;
long long sum;
int main()
{
n=read();
for(int i=1;i<=n;i++) a[i]=read(),prev[i]=i-1,next[i]=i+1;
for(int i=n;i;i--)
{
x=a[i];
u[x]=prev[x],v[x]=next[x];
prev[next[x]]=prev[x],next[prev[x]]=next[x];
}
for(int i=1;i<=n;i++)
{
x=a[i];
if((u[x]>=1)&&(u[x]<=n)) deep[x]=max(deep[x],deep[u[x]]+1);
if((v[x]>=1)&&(v[x]<=n)) deep[x]=max(deep[x],deep[v[x]]+1);
sum+=deep[x];
printf("%lld\n",sum);
}
return 0;
}
//插进来的时候 找已经插入的节点中的u和v 做完之后用p n 数组删除他 恢复成原来的状态
//所以要倒着做 当后面节点处理后,就“删除”, 因为在他插入时,前面的点都早插完了
//注意prev next的用法,开始时就定义成了i-1,i+1即最小最大元