剑指 Offer 07. 重建二叉树
1、题目
输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
2、解析
知识点:
- 前序遍历列表:第一个元素永远是 【根节点 (root)】
- 中序遍历列表:根节点 (root)【左边】的所有元素都在根节点的【左分支】,【右边】的所有元素都在根节点的【右分支】
算法思路:
- 通过【前序遍历列表】确定【根节点(root)】
- 将【中序遍历列表】的节点分割成【左分支节点】和【右分支节点】
- 递归寻找【左分支节点】中的【根节点 (left child)】和 【右分支节点】中的【根节点 (right child)】
对于以下数据:
preorder = [3,9,6,10,20,15,7] # 先序遍历
inorder = [6,9,10,3,15,20,7] # 中序遍历
- 根据[先序遍历],可以确定【3】为根节点,然后根据【3】在[中序遍历]进行划分:
- [6, 9, 10] 是【左子树】
- [3] 是此轮的【根节点】
- [15, 20, 7] 是【右子树】
- 对【左子树】进行同样的操作,递归传入的数组:
- preorder = [9, 6, 10] # 先序遍历
- inorder = [6, 9, 10] # 中序遍历
- 对【右子树】进行同样的操作,递归传入的数组:
- preorder = [20, 15, 7] # 先序遍历
- inorder = [15, 20, 7] # 中序遍历
3、代码
public TreeNode buildTree(int[] preorder, int[] inorder) {
TreeNode root = new TreeNode();
int n = preorder.length;
if(n == 0) return null;
int rootVal = preorder[0];
root.val = rootVal;
int rootIndex = 0;
for(int i = 0;i < n;i++){
if(inorder[i] == rootVal){
rootIndex = i;
break;
}
}
root.left = buildTree(Arrays.copyOfRange(preorder, 1, 1 + rootIndex), Arrays.copyOfRange(inorder,0,rootindex));
root.right=buildTree(Arrays.copyOfRange(preorder,1+rootindex,n), Arrays.copyOfRange(inorder,rootindex+1,n));
return root;
}
剑指 Offer 31. 栈的压入、弹出序列
1、题目
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如,序列 {1,2,3,4,5} 是某栈的压栈序列,序列 {4,5,3,2,1} 是该压栈序列对应的一个弹出序列,但 {4,3,5,1,2} 就不可能是该压栈序列的弹出序列。
2、题解
- 给定一个压入序列 pushedpushed 和弹出序列 poppedpopped ,则压入 / 弹出操作的顺序(即排列)是 唯一确定 的。
- 考虑借用一个辅助栈 stackstack ,模拟 压入 / 弹出操作的排列。根据是否模拟成功,即可得到结果。
- 入栈操作:按照压栈序列的顺序执行。
- 出栈操作: 每次入栈后,循环判断 “栈顶元素 == 弹出序列的当前元素” 是否成立,将符合弹出序列顺序的栈顶元素全部弹出。
3、代码
public boolean validateStackSequences(int[] pushed, int[] popped) {
Deque<Integer> stack = new ArrayDeque<Integer>();
int cur = 0;
for(int curNum : pushed){
stack.push(curNum);
while(!stack.isEmpty() && stack.peek() == popped[cur]){
stack.pop();
cur++;
}
}
return stack.isEmpty();
}