题解 | #平衡二叉树#

import java.util.*;
import java.util.Map;
public class Solution {
    public boolean IsBalanced_Solution(TreeNode root) {
        highmap.put(null,0);
        depth(root);
        return judgehighabs(root);
    }    
    Map<TreeNode,Integer> highmap=new HashMap<TreeNode,Integer>();
//求树的高度
    int depth(TreeNode root){
        if (root==null){
            return 0;            
        }
        if(highmap.containsKey(root)){
            return highmap.get(root);
        }
        int leftvalue=depth(root.left);
        int rightvalue=depth(root.right);
        highmap.put(root,Math.max(leftvalue,rightvalue)+1);
        return highmap.get(root);
    }

//判断树的作业平衡

    private boolean judgehighabs(TreeNode root){
        if(root==null){
            return true;
        }
       return (Math.abs(highmap.get(root.left)-highmap.get(root.right))<2)
                     &&judgehighabs(root.left)&&judgehighabs(root.right);

    }
}

```

全部评论

相关推荐

不愿透露姓名的神秘牛友
11-26 15:46
已编辑
字节国际 电商后端 24k-35k
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务