LeetCode hot100-45-N

Aloha_up2024-05-20 19:41

java 复制代码

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

这题做了做错了，但是参考别人的思路写了个还行的解法。

注意，从右边看的节点并不是一直递归root.right，因为某一层可能最右边的节点是空的，就看到更左边的一个节点去了。

思路就是深度递归，前序遍历。新的一层就放节点进list，但是并没有保证放进去的是最右边的节点，于是就在递归的时候一直更新同一层的那个节点。

java 复制代码

class Solution {
    List<Integer> list = new ArrayList<Integer>();

    public List<Integer> rightSideView(TreeNode root) {
        rightHelp(root,0);
        return list;

    }

    public void rightHelp(TreeNode root, int depth) {
        if (root == null) {
            return;
        }
        if(list.size()<=depth){
            list.add(root.val);
        } else {
            list.set(depth,root.val);
        }
        rightHelp(root.left, depth+1);
        rightHelp(root.right, depth+1);

    }
}