zoukankan      html  css  js  c++  java
  • 剑指Offer-- 平衡二叉树*

    题目 输入一棵二叉树,判断该二叉树是否是平衡二叉树。
    思路 递归判断,先判断自身左右子树高度差,再递归判断左右子树是否为平衡二叉树;

    package solution;
    
    public class Solution28 {
    	
    	 public boolean IsBalanced_Solution(TreeNode root) {
    	        if(root==null) { //递归出口;
    	        	return true;
    	        }
    	       
    	        return (Math.abs(getTreeHeight(root.left)-getTreeHeight(root.right))<=1)&&IsBalanced_Solution(root.left)&&IsBalanced_Solution(root.right);
    	    }
    		private int getTreeHeight(TreeNode root) {
    			if (root == null) {
    				return 0;
    			}
    			return 1 + Math.max(getTreeHeight(root.left), getTreeHeight(root.right));
    		}
    }
    			
    

    这种做法有很明显的问题,在判断上层结点的时候,会多次重复遍历下层结点,增加了不必要的开销。如果改为从下往上遍历,如果子树是平衡二叉树,则返回子树的高度;如果发现子树不是平衡二叉树,则直接停止遍历,这样至多只对每个结点访问一次。
    更巧妙的做法 求高度的同时判断是否为平衡二叉树

    链接:https://www.nowcoder.com/questionTerminal/8b3b95850edb4115918ecebdf1b4d222
    来源:牛客网
    
    public class Solution {
        public boolean IsBalanced_Solution(TreeNode root) {
            return getDepth(root) != -1;
        }
         
        private int getDepth(TreeNode root) {
            if (root == null) return 0;
            int left = getDepth(root.left);
            if (left == -1) return -1;
            int right = getDepth(root.right);
            if (right == -1) return -1;
            return Math.abs(left - right) > 1 ? -1 : 1 + Math.max(left, right);
        }
    }
    
    多思考,多尝试。
  • 相关阅读:
    qemu+chroot构建arm aarch64虚拟机
    <转>Linux环境下段错误的产生原因及调试方法小结
    <转>PCA的数学原理
    博客分类整理
    detectron2 配置记录
    如何读取部分的预训练模型
    重新配置语义分割实验环境遇到的坑
    pytorch 调整tensor的维度位置
    seg代码配置的踩坑记录
    Alienware R8外星人台式机安装双系统(WIN10+Ubuntu)的总结
  • 原文地址:https://www.cnblogs.com/LynnMin/p/9373750.html
Copyright © 2011-2022 走看看