题解 | #二叉搜索树的第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整型
     */
    TreeNode* ans=NULL;
    int KthNode(TreeNode* p, int k) {
        // write code here
        if(!p||k==0)return -1;
        KthNodeCore(p, k);
        return ans ?ans->val:-1;
    }
    void KthNodeCore(TreeNode *p, int &k)
    {
        if(!p) return;
        KthNodeCore(p->left, k);
        if(k==0)return;
        if(--k==0)
        {
            ans=p;
            return;
        }
        KthNodeCore(p->right, k);
    }
};