333. Largest BST Subtree

时间:2021-12-31 15:29:50

nlgn就不说了。。说n的方法。

这个题做了好久。

一开始想到的是post-order traversal.

左右都是BST,然后自己也是BST,返还长度是左+右+自己(1)。

左右其中一个不是,或者自己不是的话,返还-1.

第一次尝试忽略了跳级问题,判断自己能不能和左右子树组成BST的情况是,自己必须比左子树的最大值大,比右子树的最小值小。。所以害的自己造个变量记录最大最小值。

提交的时候改了半天,主要问题在于更新最大最小但是会后,既要以子树更新,又要以自己的VAL更新,一开始忽略了其中一种。。

public class Solution
{ public class MaxMin
{
int max = Integer.MIN_VALUE;
int min = Integer.MAX_VALUE;
public MaxMin(int max, int min)
{
this.max = max;
this.min = min;
}
} int a = 1;
public int largestBSTSubtree(TreeNode root)
{ if(root == null) return 0;
MaxMin r = new MaxMin(Integer.MIN_VALUE,Integer.MAX_VALUE);
int res = helper(root,r); a = Math.max(res,a); return a; } public int helper(TreeNode root, MaxMin m)
{
if(root == null) return 0;
if(root.left == null && root.right == null)
{
m.max = Math.max(m.max,root.val);
m.min = Math.min(m.min,root.val);
return 1;
} MaxMin m1 = new MaxMin(m.max,m.min);
MaxMin m2 = new MaxMin(m.max,m.min); int left = helper(root.left,m1);
int right = helper(root.right,m2);
int res = 1; m.max = Math.max(m.max,root.val);
m.min = Math.min(m.min,root.val); m.max = Math.max(Math.max(m.max,m1.max),m2.max);
m.min = Math.min(Math.min(m.min,m1.min),m2.min); if(left != -1 && right != -1)
{ if(root.val < m2.min && root.val > m1.max)
{
res += left + right;
a = Math.max(res,a);
return res;
}
else return -1; }
else return -1; } }

另一种N的尝试是,做一个in-order traversal,找其中的递增组合。不过仔细想想犹豫结构不确定,不一定是completed tree所以无法判定。。



二刷。

这个题做了他妈好久。

回头看一刷觉得太繁琐了,肯定有问题,但是又不想在基础上改。

像题目说的,直白的做法是每个NODE都call一次valid BST,这样是O(nlgn).

现在要求O(n),只能遍历一次。那肯定是bottom-up,用post-order traversal.

有一点要注意的是,subtree得到底才行,必须包含最下的leaves.

我们需要传到上面的信息有这么几个:

  1. 我是不是BST,以便于上面判断他是不是BST.

  2. 我作为BST的最大Size,以便于上面进行左右取舍,或相加(如果左右+自己都是BST)

  3. 我的取值区间,最大值和最小值。对于上面的parent node来说,他得比左支最大值大,比右支最小值小。

需要传递的信息有3个,自己建个新的class比较方便。

学到的一个tricky是用size正负来表示1),是否是BST。负数就说明我不是BST。

Time: O(n) DFS post-order

Space: O(n) for Stack in memory

public class Solution {
public class Node {
int max;
int min;
int num;
public Node(int max, int min, int num) {
this.max = max;
this.min = min;
this.num = num;
}
} public int largestBSTSubtree(TreeNode root) {
return Math.abs(dfs(root).num);
} public Node dfs(TreeNode root) {
if (root == null) return new Node(Integer.MIN_VALUE, Integer.MAX_VALUE, 0); Node leftRes = dfs(root.left);
Node rightRes = dfs(root.right); int curMin = Math.min(root.val, leftRes.min);
int curMax = Math.max(root.val, rightRes.max); if (root.val <= leftRes.max || root.val >= rightRes.min || leftRes.num < 0 || rightRes.num < 0) {
//System.out.println(leftRes.num + " " + rightRes.num);
return new Node(curMax, curMin, Math.max(Math.abs(leftRes.num), Math.abs(rightRes.num)) * -1);
} else {
return new Node(curMax, curMin, leftRes.num + rightRes.num + 1);
} }
}

2个判断不太好理解,着重说一下。

    if (root.val <= leftRes.max || root.val >= rightRes.min || leftRes.num < 0 || rightRes.num < 0) {
return new Node(curMax, curMin, Math.max(Math.abs(leftRes.num), Math.abs(rightRes.num)) * -1);
}

这说明不符合BST规定。

root.val <= 左支最大值. root.val >= 右支最小值.

左支BST的SIZE < 0 或者 右支BST的Size < 0

这个时候返还的Node要注意。

curMax和curMin说实话其实无所谓,返还什么都行,因为第三个值要返还负数。

第三个值首先要确定返还负数,最后乘以-1。

判断前取绝对值Math.abs(leftRes.num)是代表:

子树最大subTree的size,不一定就是root.left的值,可能还要往下。

正负号代表当前子树(root.left)是否是BST.

else是判断成功,左右都是,自己也是,那么显然左+右+1就行了。