leetcode230 二叉搜索树中第K小的元素 & 538 把二叉搜索树转换为累加树

labuladong

解题基础：

二叉搜索树（Binary Search Tree）特性：

• 对于 BST 的每一个节点 node，左子树节点的值都比 node 的值要小，右子树节点的值都比 node 的值大
• 对于 BST 的每一个节点 node，它的左侧子树和右侧子树都是 BST
• BST 的中序遍历结果是升序的

题目链接：

leetcode230 二叉搜索树中第K小的元素

本题思路：

/**
 * 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 int kthSmallest(TreeNode root, int k) {
        traverse(root, k);
        return res;
    }

    // 记录结果
    int res = 0;
    // 记录排名
    int rank = 0;

    // 辅助函数 遍历BST
    public void traverse(TreeNode root, int k) {
       
        // base case
        if(root == null) return;

        traverse(root.left, k);
        
        rank++;
        if(rank == k) {
            res = root.val;
            return;
        }
        
        traverse(root.right, k);
    }
}

题目链接：

leetcode538 把二叉搜索树转换为累加树

本题思路：

BST 的中序遍历代码可以升序打印节点的值，如果想降序打印节点的值，只要把递归顺序改一下就行了

class Solution {
    int sum = 0;

    public TreeNode convertBST(TreeNode root) {
        // base case
        if(root == null) return null;

        convertBST(root.right);
        
        // 维护累加和
        sum += root.val;
        // 将 BST 转化成累加树
        root.val = sum;
        
        convertBST(root.left);

        return root;
    }
}