文章目录
- 1 题目
- 2 思路
- 3 ACM模式
- 参考
1 题目
https://leetcode.cn/problems/convert-bst-to-greater-tree/description/
给出二叉 搜索 树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree)
累加树:每个节点 node 的新值等于 原树中大于或等于node.val
的值之和。
就是将原节点的值小的节点的值变为后面大的累加的值
例子
ACM模式:这里采用完全二叉树的方式构建二叉树
2 思路
原来值小的节点的新值是原来值大的节点的累加,这里需要对节点进行排序了
但是二叉搜索树又是排序好的,即中序遍历后的值就是排序好的值,左中右
二叉搜索树为升序
这里的需求又是从大的值向小的值累加,可以将中序遍历的左右顺序颠倒,即左中右
是升序排序,右中左
是降序排序;可以采用右中左
的顺序进行排序,在递归遍历的时候加入prev
指针保存累加的值
class Solution{
public TreeNode convertBST(TreeNode root) {
inorderReverseTree(root);
return root;
}
private TreeNode prev = null;
private void inorderReverseTree(TreeNode root) {
if (root != null) {
inorderReverseTree(root.right);
if (prev != null) {
root.val = root.val + prev.val;
}
prev = root;
inorderReverseTree(root.left);
}
}
}
Note: prev
指针的使用方式,先判空,非空时已经保存了之前的值,然后再赋值
3 ACM模式
import java.util.*;
class TreeNode{
int val;
TreeNode left;
TreeNode right;
TreeNode() {}
TreeNode(int val) {this.val = val; }
}
// Solution
public class Main {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
Solution solution = new Solution();
String[] nums = in.nextLine().split(" ");
List<TreeNode> nodes = new ArrayList<>();
for (String num: nums) {
if (!num.isEmpty()) {
if ("null".equals(num)) {
nodes.add(null);
} else {
nodes.add(new TreeNode(Integer.parseInt(num)));
}
}
}
constructTree(nodes);
TreeNode root = nodes.get(0);
solution.convertBST(root);
preorderTree(root);
}
// 使用满二叉树
public static void constructTree(List<TreeNode> nodes) {
for (int i = 0; 2*i+1 < nodes.size(); i++) {
if (nodes.get(i) != null) {
nodes.get(i).left = nodes.get(2*i+1);
if (2*i+2 < nodes.size()) {
nodes.get(i).right = nodes.get(2*i+2);
}
}
}
}
public static void preorderTree(TreeNode root) {
if (root != null) {
System.out.print(root.val + " ");
preorderTree(root.left);
preorderTree(root.right);
}
}
}
/*
test case:
4 1 6 0 2 5 7 null null null 3 null null null 8
*/
参考
https://programmercarl.com/0538.把二叉搜索树转换为累加树.html