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

2021-06-22 22:00:39 浏览数 (1)

给定一个二叉搜索树的根节点 root ,和一个整数 k ,请你设计一个算法查找其中第 k 个最小元素(从 1 开始计数)。

代码语言:javascript复制
class Solution {
    int  index=1;
    int ans=0;
    public int kthSmallest(TreeNode root, int k) {
        /**
        中序遍历:
            找到第k个元素返回即可
         */
         dfs(root,k);
         return ans;
      
    }

    public void dfs(TreeNode root,int k){
           if(root==null) return;

            dfs(root.left,k);         
         if(index==k){
             ans= root.val;
         }
         index  ;
         dfs(root.right,k);

    }
}

0 人点赞