Coding Interview:找到最大的搜索二叉子树

题目描述

给定一颗二叉树的头节点,找到含有节点最多的搜索二叉子树,并返回这个树的头节点。

解题思路

后续遍历这个树,当前节点的值如果大于等于左子树中的最大值,并且小于等于右子树的最小值,并且左子树是搜索二叉树,并且右子树是搜索二叉树,则当前节点是搜索二叉树。否则,选择左,右子树中最大数量的即可。

代码

/** * 找到二叉树中最大的搜索二叉树 */
public class BiggestSubTree {



    private int min;
    private int max;
    private int size;

    static class Node {
        int val;
        Node left;
        Node right;

        Node(int val){
            this.val = val;
        }
    }


    public Node biggestSubTree(Node root){
        return dfs(root);
    }

    private Node dfs(Node root){
        if(root==null){
            min = 0;
            max = 0 ;
            size = 0;
            return null;
        }else{
            Node lNode = dfs(root.left);
            int lMin = min;
            int lMax = max;
            int lSize = size;
            Node rNode = dfs(root.right);
            int rMin = min;
            int rMax = max;
            int rSize = size;
            if(lNode == null && rNode==null){
                size = 1;
                min = root.val;
                max = root.val;
                return root;
            }else if (lNode == root.left && rNode == root.right
                    && root.val >= lMax && root.val <= rMin) {
                size = lSize + rSize + 1;
                min = lMin;
                max = rMax;
                return root;
            } else {
                min = lMin;
                min = Math.min(min,root.val);
                min = Math.min(min,rMin);
                max = lMax;
                max = Math.max(max,root.val);
                max = Math.max(max,rMax);
                size = lSize>rSize?lSize:rSize;
                return lSize>rSize?lNode:rNode;
            }
        }
    }
 }
点赞