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