题解 | 二叉搜索树的第k个节点

/**
 * struct TreeNode {
 *  int val;
 *  struct TreeNode *left;
 *  struct TreeNode *right;
 *  TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param proot TreeNode类
     * @param k int整型
     * @return int整型
     */
    int inorder(TreeNode* root, int k, int& cnt) {
        if (!root) return -1;

        // 递归遍历左子树
        int left = inorder(root->left, k, cnt);
        if (left != -1) return left;

        // 访问当前节点
        cnt++;
        if (cnt == k) return root->val;

        // 遍历右子树
        return inorder(root->right, k, cnt);
    }
    int KthNode(TreeNode* proot, int k) {
        // write code here
        if (!proot || k <= 0) return -1; // 无效情况直接返回 -1
        int cnt = 0;
        return inorder(proot, k, cnt);
    }
};