题目:
给定一棵二叉树的头节点head,返回这颗树中最大的搜索二叉子树头节点
1、解决方案1
1.1、思路:
从head节点开始判断,如果以该节点为头的树是二叉搜索树,那么直接返回大小。否则就继续判断该节点的左右子树(左右子树中取最大值)
1.2、代码:
public static class Node {
public int value;
public Node left;
public Node right;
public Node(int data) {
this.value = data;
}
}
//看看以该节点为头的子树是不是二叉搜索树
//如果是,那么返回该二叉搜索树节点的数。否则就返回0
public static int getBSTSize(Node head) {
if (head == null) {
return 0;
}
ArrayList<Node> arr = new ArrayList<>();
in(head, arr);
for (int i = 1; i < arr.size(); i++) {
if (arr.get(i).value <= arr.get(i - 1).value) {
return 0;
}
}
return arr.size();
}
//中序遍历该树,然后把遍历到的数据都加入到集合中
public static void in(Node head, ArrayList<Node> arr) {
if (head == null) {
return;
}
in(head.left, arr);
arr.add(head);
in(head.right, arr);
}
//方案1:返回最大二叉搜索子树的头节点
public static Node maxSubBSTHead1(Node head) {
if (head == null) {
return null;
}
//如果以head为头的子树是二叉搜索树,那么就直接返回该节点。其左右孩子就没必要看了
if (getBSTSize(head) != 0) {
return head;
}
//如果不是,那么看看其左右孩子是否为二叉搜索树,取其中最大的
Node leftAns = maxSubBSTHead1(head.left);
Node rightAns = maxSubBSTHead1(head.right);
return getBSTSize(leftAns) >= getBSTSize(rightAns) ? leftAns : rightAns;
}
2、解决方案2
2.1、思路:
1、向左右子树分别要信息
2、返回左子树是不是二叉搜索树,如果是二叉搜索树,那么此时的二叉搜索树的大小以及头部。以及此树上最大、最小值
3、返回右子树是不是二叉搜索树,如果是二叉搜索树,那么此时的二叉搜索树的大小以及头部。以及此树上最大、最小值
4、最后根据信息来判断
2.2、代码:
//方案2:
public static Node maxSubBSTHead2(Node head) {
if (head == null) {
return null;
}
return process(head).maxSubBSTHead;
}
public static class Info {
//以该节点为头的,最大二叉搜索子树的头部
public Node maxSubBSTHead;
//以该节点为头的,最大二叉搜索子树的节点数
public int maxSubBSTSize;
public int min;
public int max;
public Info(Node h, int size, int mi, int ma) {
maxSubBSTHead = h;
maxSubBSTSize = size;
min = mi;
max = ma;
}
}
public static Info process(Node head) {
if (head == null) {
return null;
}
//当前节点左子树信息
Info leftInfo = process(head.left);
//当前节点右子树信息
Info rightInfo = process(head.right);
int min = head.value;
int max = head.value;
Node maxSubBSTHead = null;
int maxSubBSTSize = 0;
if (leftInfo != null) {
min = Math.min(min, leftInfo.min);
max = Math.max(max, leftInfo.max);
maxSubBSTHead = leftInfo.maxSubBSTHead;
maxSubBSTSize = leftInfo.maxSubBSTSize;
}
if (rightInfo != null) {
min = Math.min(min, rightInfo.min);
max = Math.max(max, rightInfo.max);
if (rightInfo.maxSubBSTSize > maxSubBSTSize) {
maxSubBSTHead = rightInfo.maxSubBSTHead;
maxSubBSTSize = rightInfo.maxSubBSTSize;
}
}
//判断一下,以head为头的整棵树是否为二叉搜索树
if (
(leftInfo == null ? true : (leftInfo.maxSubBSTHead == head.left && leftInfo.max < head.value))
&&
(rightInfo == null ? true : (rightInfo.maxSubBSTHead == head.right && rightInfo.min > head.value))
) {
//此时以head为头的子树是二叉搜索树
maxSubBSTHead = head;
//此时大小为:左子树大小 + 右子树大小 + 1
maxSubBSTSize = (leftInfo == null ? 0 : leftInfo.maxSubBSTSize)
+ (rightInfo == null ? 0 : rightInfo.maxSubBSTSize) + 1;
}
return new Info(maxSubBSTHead, maxSubBSTSize, min, max);
}