문제 설명
이진 트리의 사전 순회 및 중순 순회에 따라 이진 트리를 복원하고 이진 트리의 오른쪽 뷰를 인쇄하십시오.
설명 입력 :
바이너리 트리의 사전 주문 및 중간 주문 순회를 입력합니다 .
출력 설명 :
출력 이진 트리의 오른쪽보기
예
예 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;
}
}
테스트를 원하는 경우 Niuke.com 링크로 직접 이동하여 테스트를 수행 할 수 있습니다.