数据结构与算法之-二叉搜索树中第K小的元素

题目

给定一个二叉搜索树的根节点 root ，和一个整数 k ，请你设计一个算法查找其中第 k ****个最小元素（从 1 开始计数）。

示例 1：

输入：root = [3,1,4,null,2], k = 1
输出：1

思路

1. 首先了解什么是二叉搜索树，所谓二叉搜索树是指，左节点小于根节点，右节点大于根节点
2. 二叉树的中序遍历：左→根→右。中序遍历时将节点放进list中，最终会得到一个有序的list
3. 根据 index = k-1 取出list中的元素即为目标元素。

实现代码

/**
 * 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) {
         ArrayList<Integer> list = new ArrayList<Integer>();
         // 中序遍历
         inorder(root,list);
         return list.get(k-1);
    }

    public void inorder(TreeNode root,ArrayList<Integer> list) {
        if(root == null) return;
        inorder(root.left,list);
        list.add(root.val);
        inorder(root.right,list);
    }
}