538-把二叉搜索树转换为累加树
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如:
输入: 二叉搜索树:
5
/ 2 13
输出: 转换为累加树:
18
/ 20 13
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/convert-bst-to-greater-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
int sum;
public TreeNode convertBST(TreeNode root) {
// 逆中序遍历,求和
sum = 0;
scan(root);
return root;
}
// 递归方法
private void scan(TreeNode root) {
if(root == null) {
return;
}
scan(root.right);
sum += root.val;
root.val = sum;
scan(root.left);
}
整理一下得到:(更耗内存,因为递归方法有返回值)
int sum = 0;
public TreeNode convertBST(TreeNode root) {
// 中序遍历,求和
if(root != null) {
convertBST(root.right);
sum += root.val;
root.val = sum;
convertBST(root.left);
}
return root;
}
原文地址:https://www.cnblogs.com/angelica-duhurica/p/12236756.html
时间: 2024-10-08 15:43:54