程序员代码面试指南刷题--第三章.找到二叉树中的最大搜索二叉子树

题目描述
给定一颗二叉树,已知其中所有节点的值都不一样,找到含有节点最多的搜索二叉子树,输出该子树总节点的数量。
搜索二叉树是指对于二叉树的任何一个节点,如果它有儿子,那么左儿子的值应该小于它的值,右儿子的值应该大于它的值。
输入描述:

第一行输入两个整数 n 和 root,n 表示二叉树的总节点个数,root 表示二叉树的根节点。

以下 n 行每行三个整数 fa,lch,rch,表示 fa 的左儿子为 lch,右儿子为 rch。(如果 lch 为 0 则表示 fa 没有左儿子,rch同理)

ps:节点的编号就是节点的值。

输出描述:

最大搜索二叉树节点个数

示例1

输入

3 2
2 1 3
1 0 0
3 0 0

输出

3

解法一:树形dp问题

import java.io.*;
import java.util.*;
public class Main{
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        br.readLine();
        TreeNode root = createTree(br);
        //确定是树形dp问题答案在左子树,右子树,本身中选
        //分析可能性左,右,中
        //确定需要的信息确定结构
        //递归求解
        ResultType res = find(root);
        System.out.println(res.maxSize);
    }
    public static ResultType find(TreeNode root){
        if(root==null){
            return new ResultType(null,0,Integer.MAX_VALUE,Integer.MIN_VALUE);
        }
        ResultType left = find(root.left);
        ResultType right = find(root.right);
        int min = Math.min(root.val,Math.min(left.min,right.min));
        int max = Math.max(root.val,Math.max(left.max,right.max));
        int maxSize = left.maxSize>right.maxSize?left.maxSize:right.maxSize;
        TreeNode maxNode = left.maxSize>right.maxSize?left.maxNode:right.maxNode;
        //判断中间情况
        if(root.left==left.maxNode&&root.right==right.maxNode&&root.val>left.max&&root.val<right.min){
            maxNode = root;
            maxSize = left.maxSize+right.maxSize+1;
        }
        return new ResultType(maxNode,maxSize,min,max);
        
    }
    //递归建树
    public static TreeNode createTree(BufferedReader br){
        try{
            String[] ss = br.readLine().trim().split(" ");
            int data = Integer.parseInt(ss[0]);
            int left = Integer.parseInt(ss[1]);
            int right = Integer.parseInt(ss[2]);
            TreeNode root = new TreeNode(data);
            if(left!=0){
                root.left = createTree(br);
            }
            if(right!=0){
                root.right = createTree(br);
            }
            return root;
        }catch(Exception e){
            return null;
        }
    }
}
class ResultType{
    TreeNode maxNode;
    int maxSize;
    int min;
    int max;
    public ResultType(TreeNode maxNode,int maxSize,int min,int max){
        this.maxNode = maxNode;
        this.maxSize = maxSize;
        this.min = min;
        this.max = max;
    }
}
class TreeNode{
    int val;
    TreeNode left;
    TreeNode right;
    public TreeNode(int val){
        this.val = val;
    }
}
发布了189 篇原创文章 · 获赞 8 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_44406146/article/details/105482924