题意: 在一棵二叉树中,有很多子树,判断是否是二叉搜索子树,二叉搜索树的大小是节点数累计合。
思路:
1 判断当前的节点的左右节点是不是二叉搜索树,并且当前节点是不是,如果是说明能合并成一个大的二叉搜索树,左右节点相加再+1
2 如果左右节点任意一个不是二叉搜索树,说明当前节点也不是二叉搜索树,直接比左右节点大小,返回最大的
Node:
1 当前节点中搜索二叉树最大节点数
2 当前节点是不是搜索二叉树
代码
package 算法.二叉树;
public class test4 {
/**
* 树节点
*/
public static class Node {
//1 当前节点总数
public int nodeNum;
//2 当前节点是否是搜索二叉树
public boolean flag;
//3 当前节点下所有节点最大值
public int max;
//4 当前节点下所有节点最小值
public int min;
public int num;
public Node left;
public Node right;
public Node(int nodeNum, boolean flag, int max, int min) {
this.nodeNum = nodeNum;
this.flag = flag;
this.min = min;
this.max = max;
}
}
public static Node process(Node node) {
if (node == null) {
return new Node(0, true, 0, 0);
}
Node l = process(node.left);
Node r = process(node.right);
//1 判断当前的节点的左右节点是不是二叉搜索树,并且当前节点是不是,如果是说明能合并成一个大的二叉搜索树,左右节点相加再+1
boolean flag = false;
if (node.nodeNum > l.max && node.nodeNum < r.min) {
flag = true;
}
//2 如果满足当前是二叉搜索树的条件
if (l.flag && r.flag && flag) {
node.flag = true;
node.nodeNum = r.nodeNum+ l.num+1;
}else{
//3 如果不满足 就直接比较两个节点的最大节点数 把之前的最大传递上去
node.flag = false;
node.nodeNum = Math.max(r.nodeNum,l.nodeNum);
}
node.max = Math.max(l.max,r.max);
node.min =Math.max(l.min,r.min);
return node;
}
public static void main(String[] args) {
process(null);
}
}