递归遍历二叉树——Acwing70.二叉搜索树的第K个结点

题目描述:
给定一棵二叉搜索树,请找出其中的第 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);
        }
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_56068397/article/details/123356923