欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

[Leetcode]平衡二叉树

程序员文章站 2022-02-19 06:33:18
...

给定一个二叉树,判断其是否是一个有效的二叉搜索树。

假设一个二叉搜索树具有如下特征:

  • 节点的左子树只包含小于当前节点的数。
  • 节点的右子树只包含大于当前节点的数。
  • 所有左子树和右子树自身必须也是二叉搜索树。

示例 1:

输入:
    2
   / \
  1   3
输出: true

示例 2:

输入:
    5
   / \
  1   4
     / \
    3   6
输出: false
解释: 输入为: [5,1,4,null,null,3,6]。
     根节点的值为 5 ,但是其右子节点值为 4 。

考虑到我们前序遍历我们的平衡二叉树,将会得到一个排序好的数组,最直接的方法,就是先序遍历,看是否符合要求(执行效率很低)

class Solution {
        public boolean isValidBST(TreeNode root) {
            //可以前序遍历,看下我们的结果是否递增序列

            List<Integer> result = new ArrayList<>();
            result = deep(root);
            System.out.println(result);
            for (int i = 0; i < result.size() - 1; i++) {
                if (result.get(i).intValue() >= result.get(i + 1).intValue())
                    return false;
            }
            return true;

        }

        public List<Integer> deep(TreeNode root) {

            List<Integer> tmp = new ArrayList<>();
            if (root != null) {
                tmp.addAll(deep(root.left));
                tmp.add(root.val);
                tmp.addAll(deep(root.right));
                return tmp;
            }
            return tmp;
        }
    }

改进的前序遍历,利用栈,保存之前的值,比较当前节点的值是否比之前的节点值大(这里使用的是stack,非递归)

public boolean isValidBST(TreeNode root) {
             //用栈实现
    		//注意我们这个val是Interger类型,如果取最小值的话,没有比这个值再小的了
            double min = -Double.MAX_VALUE;
            Stack<TreeNode> stack = new Stack<>();

            while (!stack.isEmpty() || root != null) {
                while (root != null) {
                    stack.push(root);
                    root = root.left;
                }
                root = stack.pop();
                if (root.val > min) {
                    min = root.val;
                    root = root.right;


                } else
                    return false;

            }
            return true;

        }