/\*\*
\* Definition for a binary tree node.
\* public class TreeNode {
\* int val;
\* TreeNode left;
\* TreeNode right;
\* TreeNode() {}
\* TreeNode(int val) { this.val = val; }
\* TreeNode(int val, TreeNode left, TreeNode right) {
\* this.val = val;
\* this.left = left;
\* this.right = right;
\* }
\* }
\*/
class Solution {
public TreeNode convertBST(TreeNode root) {
int sum = 0;
TreeNode cur = root;
while (cur != null) {
if (cur.right == null) {
// 没有比自己值大的孩子
// 轮到处理自己
sum += cur.val;
cur.val = sum;
// 自己也遍历完了,所以该遍历比自己小的
cur = cur.left;
} else {
// 右孩子的最左边,比自己大的最小的孩子
// 也就是自己前边最后一个遍历的节点,下一个就该是自己
TreeNode leftmostOfRight = cur.right;
while (leftmostOfRight.left != null
&& leftmostOfRight.left != cur) {
leftmostOfRight = leftmostOfRight.left;
}
if (leftmostOfRight.left == null) {
// 没有做过关联,说明第一次到这里,关联当前节点,保证遍历完比自己大的最小节点之后可以轮到自己
leftmostOfRight.left = cur;
cur = cur.right;
} else {
// 第二次遍历到这里,说明比自己大的都遍历完了
// 解除关系,还原树结构
leftmostOfRight.left = null;
// 轮到处理自己
sum += cur.val;
cur.val = sum;
// 自己也遍历完了,所以该遍历比自己小的
cur = cur.left;
}
}
}
return root;
}
}
c
/\*\*
\* Definition for a binary tree node.
\* struct TreeNode {
\* int val;
\* struct TreeNode \*left;
\* struct TreeNode \*right;
\* };
\*/
struct TreeNode\* convertBST(struct TreeNode\* root){
int sum = 0;
struct TreeNode \*cur = root;
while (cur != NULL) {
if (cur->right == NULL) {
// 没有比自己值大的孩子
// 轮到处理自己
sum += cur->val;
cur->val = sum;
// 自己也遍历完了,所以该遍历比自己小的
cur = cur->left;
} else {
// 右孩子的最左边,比自己大的最小的孩子
// 也就是自己前边最后一个遍历的节点,下一个就该是自己
struct TreeNode \*leftmostOfRight = cur->right;
while (leftmostOfRight->left != NULL
&& leftmostOfRight->left != cur) {
leftmostOfRight = leftmostOfRight->left;
}
if (leftmostOfRight->left == NULL) {
// 没有做过关联,说明第一次到这里,关联当前节点,保证遍历完比自己大的最小节点之后可以轮到自己
leftmostOfRight->left = cur;
cur = cur->right;
} else {
// 第二次遍历到这里,说明比自己大的都遍历完了
// 解除关系,还原树结构
leftmostOfRight->left = NULL;
// 轮到处理自己
sum += cur->val;
cur->val = sum;
// 自己也遍历完了,所以该遍历比自己小的
cur = cur->left;
}
}
}
return root;
}
c++
/\*\*
\* Definition for a binary tree node.
\* struct TreeNode {
\* int val;
\* TreeNode \*left;
\* TreeNode \*right;
\* TreeNode() : val(0), left(nullptr), right(nullptr) {}
\* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
\* TreeNode(int x, TreeNode \*left, TreeNode \*right) : val(x), left(left), right(right) {}
\* };
\*/
class Solution {
public:
TreeNode\* convertBST(TreeNode\* root) {
int sum = 0;
TreeNode \*cur = root;
while (cur != nullptr) {
if (cur->right == nullptr) {
// 没有比自己值大的孩子
// 轮到处理自己
sum += cur->val;
cur->val = sum;
// 自己也遍历完了,所以该遍历比自己小的
cur = cur->left;
} else {
// 右孩子的最左边,比自己大的最小的孩子
// 也就是自己前边最后一个遍历的节点,下一个就该是自己
TreeNode \*leftmostOfRight = cur->right;
while (leftmostOfRight->left != nullptr
&& leftmostOfRight->left != cur) {
leftmostOfRight = leftmostOfRight->left;
}
if (leftmostOfRight->left == nullptr) {
// 没有做过关联,说明第一次到这里,关联当前节点,保证遍历完比自己大的最小节点之后可以轮到自己
leftmostOfRight->left = cur;
cur = cur->right;
} else {
// 第二次遍历到这里,说明比自己大的都遍历完了
// 解除关系,还原树结构
leftmostOfRight->left = nullptr;
// 轮到处理自己
sum += cur->val;
cur->val = sum;
// 自己也遍历完了,所以该遍历比自己小的
cur = cur->left;
}
}
}
return root;
}
};
python
# Definition for a binary tree node.
# class TreeNode:
# def \_\_init\_\_(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def convertBST(self, root: TreeNode) -> TreeNode:
total = 0
cur = root
while cur:
if not cur.right:
# 没有比自己值大的孩子
# 轮到处理自己
total += cur.val
cur.val = total
# 自己也遍历完了,所以该遍历比自己小的
cur = cur.left
else:
# 右孩子的最左边,比自己大的最小的孩子
# 也就是自己前边最后一个遍历的节点,下一个就该是自己
leftmostOfRight = cur.right
while leftmostOfRight.left and leftmostOfRight.left != cur:
leftmostOfRight = leftmostOfRight.left
if not leftmostOfRight.left:
# 没有做过关联,说明第一次到这里,关联当前节点,保证遍历完比自己大的最小节点之后可以轮到自己
leftmostOfRight.left = cur
cur = cur.right
else:
# 第二次遍历到这里,说明比自己大的都遍历完了
# 解除关系,还原树结构
leftmostOfRight.left = None
# 轮到处理自己
total += cur.val
cur.val = total
# 自己也遍历完了,所以该遍历比自己小的
cur = cur.left
return root
go
/\*\*
\* Definition for a binary tree node.
\* type TreeNode struct {
\* Val int
\* Left \*TreeNode
\* Right \*TreeNode
\* }
\*/
func convertBST(root \*TreeNode) \*TreeNode {
sum := 0
cur := root
for cur != nil {
if cur.Right == nil {
// 没有比自己值大的孩子
// 轮到处理自己
sum += cur.Val
cur.Val = sum
// 自己也遍历完了,所以该遍历比自己小的
cur = cur.Left
} else {
// 右孩子的最左边,比自己大的最小的孩子
// 也就是自己前边最后一个遍历的节点,下一个就该是自己
leftmostOfRight := cur.Right
for leftmostOfRight.Left != nil && leftmostOfRight.Left != cur {
leftmostOfRight = leftmostOfRight.Left
}
if leftmostOfRight.Left == nil {
// 没有做过关联,说明第一次到这里,关联当前节点,保证遍历完比自己大的最小节点之后可以轮到自己
leftmostOfRight.Left = cur
cur = cur.Right
} else {
// 第二次遍历到这里,说明比自己大的都遍历完了
// 解除关系,还原树结构
leftmostOfRight.Left = nil
// 轮到处理自己
sum += cur.Val
cur.Val = sum
// 自己也遍历完了,所以该遍历比自己小的
cur = cur.Left
}
}
}
return root
}