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);
}
}