问题描述
请根据二叉树的前序遍历,中序遍历恢复二叉树,并打印出二叉树的右视图。
输入描述:
输入二叉树的前序和中序遍历
输出描述:
输出二叉树的右视图
示例
示例1
输入
[1,2,4,5,3],[4,2,5,1,3]
输出
[1,3,5]
解决思路
分析
- 根据二叉树的前序和中序还原二叉树,然后通过广度优先遍历得到二叉树的右视图
方法
- 根据二叉树的前序和中序遍历的特点还原二叉树(可以看考这篇文章二叉树的重建)
- 通过队列实现二叉树的广度优先遍历,得到最右侧的节点,从而得到右视图。
代码实现
// 思路1
public class Solution {
public int[] solve(int[] pre, int[] in) {
// write code here
TreeNode root = recoverTree(pre, in);
List<Integer> list = getRightView(root);
int[] res = new int[list.size()];
for (int i = 0; i < list.size(); i++) {
res[i] = list.get(i);
}
return res;
}
// 重构二叉树
public TreeNode recoverTree(int[] pre, int[] in) {
if (pre.length == 0 || in.length == 0) {
return null;
}
// 前序第一个节点为跟节点
TreeNode root = new TreeNode(pre[0]);
// 遍历中序找跟节点
for (int i = 0; i < in.length; i++) {
if (pre[0] == in[i]) {
root.left = recoverTree(Arrays.copyOfRange(pre, 1, i + 1), Arrays.copyOfRange(in, 0, i));
root.right = recoverTree(Arrays.copyOfRange(pre, i + 1, pre.length), Arrays.copyOfRange(in, i + 1, in.length));
}
}
return root;
}
// 查看右视图,右视图即二叉树从右向左看的视图
// 采用广度优先搜索遍历,搜索每一层,只保留该层最后一个节点
public List<Integer> getRightView(TreeNode root) {
// 通过链表实现队列
LinkedList<TreeNode> list = new LinkedList<>();
// 存储右视图
List<Integer> res = new ArrayList<>();
// 添加跟节点
list.add(root);
while (!list.isEmpty()) {
// size主要用于记录每一层的节点数,方便跳出循环遍历
int size = list.size();
TreeNode cur = null;
while (size > 0) {
size--;
// 从队列的第一个节点开始取(即从左向右遍历获取,最后一个节点为右视图)
cur = list.removeFirst();
// 向队列中添加左节点
if (cur.left != null) {
list.add(cur.left);
}
// 向队列中添加右节点
if (cur.right != null) {
list.add(cur.right);
}
}
// 节点的最后一个值方法右视图
res.add(cur.val);
}
// 返回右视图
return res;
}
}
小伙伴如果想测试的话,可以直接到牛客网这个链接做测试