给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如:
输入: 二叉搜索树:
5
/ \
2 13
输出: 转换为累加树:
18
/ \
20 13
解法:
/**
* Definition for a binary tree node.
* public class TreeNode {
* public var val: Int
* public var left: TreeNode?
* public var right: TreeNode?
* public init(_ val: Int) {
* self.val = val
* self.left = nil
* self.right = nil
* }
* }
*/
class Solution {
var num = 0
func convertBST(_ root: TreeNode?) -> TreeNode? {
if (root != nil) {
//遍历右子树
convertBST(root!.right);
//遍历顶点
root!.val = root!.val + num;
num = root!.val;
//遍历左子树
convertBST(root!.left);
return root;
}
return nil;
}
}