leetcode-230. 二叉搜索树中第K小的元素

dfs 中序遍历即可

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
 
var num, ans int
var find bool
 
 
func kthSmallest(root *TreeNode, k int) int {
    find = false
    num, ans = 0, 0
 
    dfs(root, k)
    return ans
}
 
func dfs(root *TreeNode, k int) {
    if find || root == nil {
        return
    }
 
    if root.Left != nil {
        dfs(root.Left, k)
    }
 
    num++
 
    if num == k {
        ans = root.Val
        find = true
    }
 
    if root.Right != nil {
        dfs(root.Right, k)
    }
 
    return
}

参考

230. 二叉搜索树中第K小的元素 - 力扣（Leetcode）

题解中有后续优化的解法

题解-230. 二叉搜索树中第K小的元素 - 力扣（Leetcode）