题目
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如:
输入: 原始二叉搜索树:
5
/
2 13
输出: 转换为累加树:
18
/
20 13
关于二叉搜索树
开始看题目的时候,没注意二叉搜索树的定义
导致第一次实现代码如下:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode convertBST(TreeNode root) {
dfs(root);
return root;
}
public void dfs(TreeNode root){
错误认为比较当前递归三个节点的值的大小,然后找大值求和重新赋值即可
int roo=0,left=0,right=0;
if(root==null){
return;
}
if(root.left!=null){
if(root.left.val>root.val){
roo+=root.left.val;
}
else{
left+=root.val;
}
if(root.right!=null){
if(root.right.val>root.left.val){
left+=root.right.val;
}
else{
right+=root.left.val;
}
}
}
if(root.right!=null){
if(root.right.val>root.val){
roo+=root.right.val;
}
else{
right+=root.val;
}
}
root.val+=roo;
if(root.left!=null){
root.left.val+=left;
}
if(root.right!=null){
root.right.val+=right;
}
dfs(root.left);
dfs(root.right);
}
}
上述代码实际测试时,[2,0,3,-4,1]会出现第三层的右叶子节点无法加二层比其大的值的和
事实上,二叉搜索树定义如下
二叉搜索树是一棵空树,或者是具有下列性质的二叉树:
1. 若它的左子树不空,则左子树上所有节点的值均小于它的根节点的值;
2. 若它的右子树不空,则右子树上所有节点的值均大于它的根节点的值;
3. 它的左、右子树也分别为二叉搜索树。
由这样的性质我们可以发现,二叉搜索树的中序遍历是一个单调递增的有序序列。如果我们反序地中序遍历该二叉搜索树,即可得到一个单调递减的有序序列。
因此,正确代码如下:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
int roo=0;
public TreeNode convertBST(TreeNode root) {
dfs(root);
return root;
}
public void dfs(TreeNode root){
if(root==null){
return;
}
dfs(root.right);//反中序遍历,即值递减排列
root.val+=roo;//roo需要为全局变量,本层递归root.val与比自己大的节点值求和
roo=root.val;//roo记录本次的大节点值供下次使用,
//本次大节点值已经求过和,下次直接加roo即可,即root.val+=roo;
dfs(root.left);
}
}