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

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

给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。

说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。

示例 1:

输入: root = [3,1,4,null,2], k = 1
   3
  / \
 1   4
  \
   2
输出: 1

示例 2:

输入: root = [5,3,6,2,4,null,null,1], k = 3
       5
      / \
     3   6
    / \
   2   4
  /
 1
输出: 3

进阶：
如果二叉搜索树经常被修改（插入/删除操作）并且你需要频繁地查找第 k 小的值，你将如何优化 kthSmallest 函数？

---

思路：非递归中序遍历

    func kthSmallest(root *TreeNode, k int) int {
    	stack := make([]*TreeNode, 0)
    	var node *TreeNode
    	for {
    
    		// 遍历左子树
    		for root != nil {
    			stack = append(stack, root)
    			root = root.Left
    		}
    
    		// 遍历根节点
    		if len(stack) == 0 {
    			break
    		}
    
    		// 遍历右子树
    		stack, node = stack[:len(stack)-1], stack[len(stack)-1]
    		k--
    		if k == 0 {
    			return node.Val
    		}
    		root = node.Right
    
    	}
    	return -1
    }