【CF 675D Tree Construction】BST

简介: 题目链接:http://codeforces.com/problemset/problem/675/D 题意:给一个由n个互异整数组成的序列a[],模拟BST的插入过程,依次输出每插入一个元素a[i]后a[i]的父节点。

题目链接:http://codeforces.com/problemset/problem/675/D

题意:给一个由n个互异整数组成的序列a[],模拟BST的插入过程,依次输出每插入一个元素a[i]后a[i]的父节点。

数据范围:n [2, 10^5]

思路:直接模拟一般的BST而不维护平衡性的话,有可能会出现极度不平衡甚至退化的情况,复杂度会从O(nlogn)上升到O(n^2)。因此要用平衡二叉树。

可以利用STL中的set容器,但对于题目所要找的“父节点”,set并不提供接口。这时就要考察BST的一些性质推导出解法。

回顾二叉树的中序遍历,对节点prev的直接后继succ的定位操作分两种情况。注意等价BST的“上下可变,左右不乱”的性质,不论是否进行了等价变换,中序遍历序列中任意两个互为直接前驱和直接后继的元素,其层次关系必然为如下两种之一:

1. succ层次更深

=> 由顺序性,succ必为prev的右子树中的节点,故prev的右子树必非空

  且由“直接性”,succ的左孩子必为空。

  对于v小于全局最小值的边界情况,prev及其左子树为空。

2. prev层次更深

=> 由顺序性,prev必为succ的左子树中的节点,故succ的左子树必非空

  且由“直接性”,prev的右孩子必为空。

  对于v大于全局最大值的边界情况,succ及其右子树为空。

 

对于一个新的待插入的节点v,我们在当前BST的中序遍历序列 s 中进行二分查找,得到应插入的位置的后继元素的位置succ(“大于v的第一个元素,即upper_bound”),然后得到prev=succ-1。为了保证BST的顺序性,v必然要插在prev和succ之间。

从树的结构上看,可以插在标有“必为空”的位置,它恰好介于prev和succ之间,顺序性必然得到保证。具体地,即“succ和prev中更深的那个”,1、2两种情况分别对应succ和prev。如何确定是哪种情况呢?

 

我们回到对这两种情况的描述上:刚刚所做的推导是否可逆呢?如果可逆,那么我们可以通过判断succ或prev的左右孩子是否为空就可以得知是哪种情况。

分析发现,确实可逆:

1. succ的左孩子为空 => 由顺序性,prev必为succ的祖先 => succ层次更深。

2. prev的右孩子为空 => 由顺序性,succ必为prev的祖先 => prev层次更深。

 

到此,可以着手设计算法了。首先用set维护平衡二叉树,每次插入节点v前,调用set的lower_bound(或upper_bound,元素互异故二者无差别) 得到“大于v的第一个元素”,即插入v后v的直接后继,记录为迭代器succ。然后得到succ的直接前驱的迭代器prev = succ - 1。

对于左右孩子情况的记录,我没有想到方法,CF题解给出的是维护两个map<int, int>left, right,left记录节点对<v, lc>,right记录节点对<v, rc>。每次插入前通过判断left[succ]和right[prev]是否为空来判断父节点是谁,以及v作为左孩子还是右孩子插入,更新map。

其实这两种情况是对立的,因此一次判断就可确定属于哪种。

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <set>
 4 #include <map>
 5 using namespace std;
 6 const int MAX_N = 100005;
 7 
 8 int n;
 9 struct Node
10 {
11     int d;
12     int lc, rc;
13     Node(){}
14     Node(int d):d(d), lc(-1), rc(-1){}
15 }nodes[MAX_N];
16 
17 int a[MAX_N];
18 
19 int main()
20 {
21     while(~scanf("%d", &n)){
22         for(int i=0; i<n; i++){
23             scanf("%d", &a[i]);
24         }
25         set<int> s;
26         map<int, int> left;//<节点,左孩子> 
27         map<int, int> right; //<节点,右孩子> 
28         int res;
29         s.insert(a[0]);
30         for(int i=1; i<n; i++){
31             set<int>::iterator pos = s.lower_bound(a[i]);//直接后继 
32             if(pos != s.end() && left.count(*pos)==0){//后继没有左孩子,插到后继的左孩子位置
33                 res = *pos;
34                 left[*pos] = a[i];
35             }else{//后继有左孩子,或没有后继,插到前驱的右孩子位置 
36                 pos--;
37                 res = *pos;
38                 right[*pos] = a[i];
39             }
40             printf("%d ", res);
41             s.insert(a[i]);
42         }
43         printf("\n");
44     }
45     return 0;
46 }

p.s: CF题解的代码好优美,学习了。

目录
相关文章
|
5月前
Leetcode 236. Lowest Common Ancestor of a Binary Tree
根据LCA的定义,二叉树中最小公共祖先就是两个节点p和q最近的共同祖先节点,LCA的定义没什么好解释的,主要是这道题的解法。
15 0
LeetCode 105. Construct Binary Tree
给定一颗二叉树的前序和顺序遍历,构造原二叉树。 注意:您可以假设树中不存在重复项。
34 0
LeetCode 105. Construct Binary Tree
LeetCode 106. Construct Binary Tree
给定一颗二叉树的中序和后续遍历,构造原二叉树。 注意:您可以假设树中不存在重复项。
48 0
LeetCode 106. Construct Binary Tree
LeetCode 110. Balanced Binary Tree
给定一颗二叉树,判断此树是不是平衡二叉树. 平衡二叉树的条件是:当前节点的左右子树高度差不超过1,且左右子树都是平衡二叉树.
47 0
LeetCode 110. Balanced Binary Tree
|
机器学习/深度学习
Data Structures (五) - 二叉树Binary Tree
Data Structures (五) - 二叉树Binary Tree
Data Structures (五) - 二叉树Binary Tree
PAT (Advanced Level) Practice - 1138 Postorder Traversal(25 分)
PAT (Advanced Level) Practice - 1138 Postorder Traversal(25 分)
82 0
[LeetCode]--235. Lowest Common Ancestor of a Binary Search Tree
Given a binary search tree (BST), find the lowest common ancestor (LCA) of two given nodes in the BST. According to the definition of LCA on Wikipedia: “The lowest common ancestor is defin
1209 0