题目地址:
https://leetcode.com/problems/maximum-binary-tree-ii/
考虑Treap的建树方式,给定一个数组 A A A,该Treap的堆性质由 A [ i ] A[i] A[i]来维护,其BST性质由下标维护。那么Treap的建树方式的递归的,树根是全数组最大值 A [ k ] A[k] A[k],然后其左子树和右子树分别是 A [ 0 : k − 1 ] A[0:k-1] A[0:k−1]和 A [ k + 1 : ] A[k+1:] A[k+1:],接着就可以递归建树了。给定一棵已经建好的Treap的树根,其是以某个数组 A A A建立的(该数组事实上是唯一的),要求返回 A ∪ { x } A\cup \{x\} A∪{x}建立的Treap的树根,直接在给出的Treap里把 x x x加进去,不允许新建树。
如果 x x x自己就是最大值,那么整棵树都应该是 x x x的左子树,则new出值为 x x x的节点,然后将原树作为左子树接上去;否则的话, x x x必然应该填进右子树里,递归求解即可。代码如下:
public class Solution {
public TreeNode insertIntoMaxTree(TreeNode root, int val) {
TreeNode node = new TreeNode(val);
if (root == null) {
return node;
}
if (root.val < val) {
node.left = root;
return node;
}
root.right = insertIntoMaxTree(root.right, val);
return root;
}
}
class TreeNode {
int val;
TreeNode left, right;
public TreeNode(int val) {
this.val = val;
}
}
时空复杂度 O ( h ) O(h) O(h)。