构造MaxTree
对于一个没有重复元素的整数数组,请用其中元素构造一棵MaxTree,MaxTree定义为一棵二叉树,其中的节点与数组元素一一对应,同时对于MaxTree的每棵子树,它的根的元素值为子树的最大值。现有一建树方法,对于数组中的每个元素,其在树中的父亲为数组中它左边比它大的第一个数和右边比它大的第一个数中更小的一个。若两边都不存在比它大的数,那么它就是树根。请证明这个方法的正确性,同时设计O(n)的算法实现这个方法。
给定一个无重复元素的数组A和它的大小n,请返回一个数组,其中每个元素为原数组中对应位置元素在树中的父亲节点的编号,若为根则值为-1。
测试样例:
[3,1,4,2],4
返回:[2,0,-1,2]
import java.util.*;
public class MaxTree {
public int[] buildMaxTree(int[] A, int n) {
int res[] = new int[n];
Stack<Integer> s = new Stack<>();
for(int i = 0; i < n; i++){
int pos = -1;
//使队列保持递减(递减的都放进去),队列中存储索引
while(!s.empty() && A[s.peek()] < A[i]){
int t = s.pop();
// 此时在右边出现大的元素,需要对队列中的元素的父节点进行更新
if(res[t] == -1 || A[res[t]] > A[i])
res[t] = i;
}
if(!s.empty())
pos = s.peek();
s.push(i);
res[i] = pos;
}
return res;
}
}
拓扑结构相同子树
对于两棵彼此独立的二叉树A和B,请编写一个高效算法,检查A中是否存在一棵子树与B树的拓扑结构完全相同。
给定两棵二叉树的头结点A和B,请返回一个bool值,代表A中是否存在一棵同构于B的子树。
import java.util.*;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}*/
public class IdenticalTree {
public boolean chkIdentical(TreeNode A, TreeNode B) {
if (A == null && B != null) { //这句不能掉,否则可能会抛出空指针异常
return false;
}
if (isIdentical(A, B)) {
return true;
}
if (chkIdentical(A.left, B)) {
return true;
} else if (chkIdentical(A.right, B)) {
return true;
}
return false;
}
public boolean isIdentical(TreeNode A, TreeNode B) {
if (A == null && B == null) {
return true;
}
if (A == null || B == null) {
return false;
}
return A.val == B.val && isIdentical(A.left, B.left) && isIdentical(A.right, B.right);
}
}