034、二叉搜索树特性篇（labuladong）

东哥带你刷二叉搜索树（特性篇）

基于labuladong的算法网站，东哥带你刷二叉搜索树（特性篇）

1、BST

BST：binary search tree，二叉搜索树。
特性：

• 对于BST的每一个node节点来说，左子树的节点值都比node小，右子树的节点值都比node大；
• 对于BST的每一个节点node，它的左侧子树和右侧子树都是BST。

BST的中序遍历结果是升序的

2、寻找第K小的元素

力扣第230题，二叉搜索树中第K小的元素

[230]二叉搜索树中第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) {
        LinkedList<Integer> arr = new LinkedList<>();
        traverse(root, arr);
        // 开始遍历链表寻找到倒数第K个元素，采用双指针遍历寻找到第K个小的元素
        return arr.get(k - 1);
    }

    // 中序遍历一个二叉搜索树，将中序遍历结果放入一个链表中
    void traverse(TreeNode root, List<Integer> arr) {
        // base case
        if (root == null) {
            return;
        }
        traverse(root.left, arr);
        arr.add(root.val);
        traverse(root.right, arr);
    }
}

这是基本思路，还可以定义两个变量：

class Solution {
    public int kthSmallest(TreeNode root, int k) {
        traverse(root, k);
        return res;
    }

    int count = 0;
    int res = 0;

    // 中序遍历一个二叉搜索树
    void traverse(TreeNode root, int k) {
        // base case
        if (root == null) {
            return;
        }
        traverse(root.left, k);
        count++;
        if (count == k) {
            res = root.val;
            return;
        }
        traverse(root.right, k);
    }
}

3、BST转化累加树

力扣第538题和力扣第1038题，把二叉搜索树转换为累加树

[538]把二叉搜索树转换为累加树

/**
 * 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 TreeNode convertBST(TreeNode root) {
        traverse(root);
        return root;
    }

    int res = 0;

    void traverse(TreeNode root) {
        if (root == null) {
            return;
        }
        // 右
        traverse(root.right);

        // 中
        res += root.val;
        root.val = res;
        // 左
        traverse(root.left);
    }
}