剑指Offer打卡day43——Acwing 70. 二叉搜索树的第k个结点

Posted 2021-05-25 Johnny*

【题目描述】

Acwing 70. 二叉搜索树的第k个结点
【思路】
二叉搜索树的中序遍历 对应升序序列
那么第k小就是遍历的第k个数

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    
    TreeNode ans;
    int k; // k 要设置成全局变量
    public void inorder(TreeNode node){
        
        if( node == null ) return;
        inorder(node.left);
        k --; 
        if( k == 0){
            ans = node;
            return;
        }
        inorder(node.right);
    }
    
    public TreeNode kthNode(TreeNode root, int _k) {
        k = _k;
        inorder(root);
        return ans;
        
    }
}