LeetCode-热题100-笔记-day29

199. 二叉树的右视图https://leetcode.cn/problems/binary-tree-right-side-view/

给定一个二叉树的 根节点 root，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。

示例 1:

输入: [1,2,3,null,5,null,4]
输出: [1,3,4]

算法思路

使用层次遍历按层输出该二叉树并保存于临时数组cur中，然后遍历cur获取每一层最后一个元素添加大ans即可得到答案；

class Solution {
    public List<Integer> rightSideView(TreeNode root) {
        List<Integer> ans=new ArrayList<>();
        List<List<Integer>> cur=new ArrayList<List<Integer>>();
        if(root==null){
          return ans;
        }
        Queue<TreeNode> queue=new ArrayDeque<>();
        queue.offer(root);
        while(!queue.isEmpty()){
          List<Integer> level=new ArrayList<>();
          int size=queue.size();
          for(int i=0;i<size;i++){
            TreeNode tmp=queue.poll();
            if(tmp.left!=null){
              queue.offer(tmp.left);
            }
            if(tmp.right!=null){
              queue.offer(tmp.right);
            }
            level.add(tmp.val);
          }
          cur.add(level);
        }
        for(List<Integer> innerList:cur){
          if (!innerList.isEmpty()) { // 检查子列表是否非空
          Integer element = innerList.get(innerList.size()-1); // 获取子列表的最后一个元素
          ans.add(element);
          }
        }
        return ans;
    }
}

 

230. 二叉搜索树中第K小的元素https://leetcode.cn/problems/kth-smallest-element-in-a-bst/

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

示例 1：

输入：root = [3,1,4,null,2], k = 1
输出：1

算法思路

中序遍历可得到顺序的二叉树元素；将遍历结果存储到列表中访问第K个元素即可； 

class Solution {
    List<Integer> ans=new ArrayList<>();
    public int kthSmallest(TreeNode root, int k) {
        dfs(root);
        return ans.get(k-1);
    }

    public void dfs(TreeNode root){
        if(root==null){
            return;
        }
        dfs(root.left);
        ans.add(root.val);
        dfs(root.right);
    }
}