【Java实现】二叉搜索树的第K个结点

题目描述

给定一棵二叉搜索树，请找出其中的第k小的结点。例如， （5，3，7，2，4，6，8） 中，按结点数值大小顺序第三小结点为4。

解题思路

这道题是找数组的第K小结点的变形题，都是先把它转换成一个升序的数组然后寻找第K个元素，但是如何把二叉搜索树转换成一个升序数组呢？我们根据二叉搜索树的特点，知道二叉搜索树的中序遍历就是一个升序数组，所以我们可以对给的二叉搜索树进行中序遍历再找到第K个结点就可以了。

代码实现

import java.util.ArrayList;
public class Solution {
    ArrayList<TreeNode> list = new ArrayList<>(); 
    TreeNode KthNode(TreeNode pRoot, int k) {
        infixOrder(pRoot);
        if(k>=1 && list.size()>=k) {
            return list.get(k-1);
        }
        return null;
    }
    // 中序遍历
   private void infixOrder(TreeNode cur) {   
        if(cur != null) {
            infixOrder(cur.left);
            list.add(cur);
            infixOrder(cur.right);
        }
    }
}

最后

以上就是有魅力鸭子最近收集整理的关于【Java实现】二叉搜索树的第K个结点的全部内容，更多相关【Java实现】二叉搜索树内容请搜索靠谱客的其他文章。