这篇文章主要介绍了LeetCode如何把二叉搜索树转换为累加树,具有一定借鉴价值,感兴趣的朋友可以参考下,希望大家阅读完这篇文章之后大有收获,下面让小编带着大家一起了解一下。
1,问题简述
给出二叉 搜索 树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree),使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。提醒一下,二叉搜索树满足下列约束条件:节点的左子树仅包含键 小于 节点键的节点。节点的右子树仅包含键 大于 节点键的节点。左右子树也必须是二叉搜索树。
2,示例
输入:[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
输出:[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]
示例 2:
输入:root = [0,null,1]
输出:[1,null,1]
示例 3:
输入:root = [1,0,2]
输出:[3,3,2]
示例 4:
输入:root = [3,2,4,1]
输出:[7,9,4,10]
提示:
树中的节点数介于 1 和 100 之间。
每个节点的值介于 0 和 100 之间。
树中的所有值 互不相同 。
给定的树为二叉搜索树。
3,题解思路
写了最基本的思路来解决,先统计二叉树的节点数据,然后根据题目的已知条件进行计算,数据赋值就可以了
4,题解程序
import java.util.ArrayList;
import java.util.List;
public class ConvertBSTTest {
static List<Integer> list = new ArrayList<>();
public static void main(String[] args) {
TreeNode t1 = new TreeNode(4);
TreeNode t2 = new TreeNode(1);
TreeNode t3 = new TreeNode(6);
TreeNode t4 = new TreeNode(0);
TreeNode t5 = new TreeNode(2);
TreeNode t6 = new TreeNode(5);
TreeNode t7 = new TreeNode(7);
TreeNode t8 = new TreeNode(3);
TreeNode t9 = new TreeNode(8);
t1.left = t2;
t1.right = t3;
t2.left = t4;
t2.right = t5;
t3.left = t6;
t3.right = t7;
t5.right = t8;
t7.right = t9;
TreeNode treeNode = convertBST(t1);
System.out.println("treeNode = " + treeNode);
}
public static TreeNode convertBST(TreeNode root) {
if (root == null) {
return null;
}
if (list.size() == 0) {
dfs(root);
}
Integer compute = compute(root.val);
root.val = compute;
if (root.left != null) {
convertBST(root.left);
}
if (root.right != null) {
convertBST(root.right);
}
return root;
}
private static Integer compute(Integer val) {
int sum = 0;
for (int num : list) {
if (num > val) {
sum += num;
}
}
sum += val;
return sum;
}
private static void dfs(TreeNode root) {
if (root == null) {
return;
}
if (root.left != null) {
dfs(root.left);
}
list.add(root.val);
if (root.right != null) {
dfs(root.right);
}
}
}
5,题解程序图片版
6
感谢你能够认真阅读完这篇文章,希望小编分享的“LeetCode如何把二叉搜索树转换为累加树”这篇文章对大家有帮助,同时也希望大家多多支持亿速云,关注亿速云行业资讯频道,更多相关知识等着你来学习!
原创文章,作者:506227337,如若转载,请注明出处:https://blog.ytso.com/223740.html