[LeetCode] 230. Kth Smallest Element in a BST 解题思路

Given a binary search tree, write a function kthSmallest to find the kth smallest element in it.

Note: 
You may assume k is always valid, 1 ≤ k ≤ BST's total elements.

 问题：找出二叉搜索树种第 k 小的元素。

一个深度遍历的应用。使用递归、或者借助栈都可以实现深度遍历。本文代码使用递归实现。

    void visit(TreeNode* node){
        
        if (node->left != NULL){
            visit(node->left);
        }
        
        if (cnt == 0) {
            return;
        }
        
        cnt--;
        if(cnt == 0){
            res = node->val;
            return ;
        }
        
        if(node->right != NULL){
            visit(node->right);
        }
    }

    int cnt;
    int res;

    int kthSmallest(TreeNode* root, int k) {
        cnt = k;
        if(root == NULL){
            return 0;
        }
        visit(root);
        
        return (cnt == 0) ? res : 0;
    }