题目描述:
给定一棵二叉搜索树,请找出其中的第 k 小的结点。
你可以假设树和 k 都存在,并且 1≤k≤ 树的总结点数。
数据范围
树中节点数量 [1,500]。
算法思路:
本题其实就是考查对于二叉树的遍历,有两种方法,一种利用栈进行非递归的遍历,另一种采用递归的方式;我才用的是递归方式进行遍历,递归遍历当前结点的左右子树,如果左右子树都为空那么就结束,否则遍历左右子树。
同时在递归方法之外,定义一个全局变量用来存储二叉树的结点,对递归遍历完的二叉树中结点的元素进行排序,取出第k小的结点即可。
算法实现:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode kthNode(TreeNode root, int k) {
//创建一个集合用来存储元素
List<TreeNode> list= new ArrayList<>();
dfs(root,list);
int[] res = new int[list.size()];
for(int i=0;i<list.size();i++){
res[i] = list.get(i).val;
}
Arrays.sort(res);
return new TreeNode(res[k-1]);
}
//定义方法来遍历二叉树
public void dfs(TreeNode root,List<TreeNode> list){
list.add(root);
//边界条件
if(root.left==null&&root.right==null){
return;
}
if(root.left!=null){
//遍历左子树
dfs(root.left,list);
}
if(root.right!=null){
//遍历右子树
dfs(root.right,list);
}
}
}