剑指第二版第55题,二叉搜索树的判定

最新推荐文章于 2024-10-12 20:00:42 发布

LIMao_Link

最新推荐文章于 2024-10-12 20:00:42 发布

阅读量52

点赞数

分类专栏：剑指Offer第二版文章标签：链表数据结构 leetcode go

本文链接：https://blog.csdn.net/qq_53736949/article/details/121397046

版权

剑指Offer第二版专栏收录该内容

20 篇文章 0 订阅

订阅专栏

剑指offer第55题:判断一个平衡二叉树

func isBalanced(root *TreeNode) bool {
   res := dfs(root) != -1
   return res
}
//先写第一种方法,后序遍历 + 剪枝
func dfs(root *TreeNode) int{
    if root == nil{ return 0}
       
    

   left_tree := dfs(root.Left)
   if left_tree == -1{
       return -1;
   }

   right_tree := dfs(root.Right)
   if right_tree == -1{
       return -1
   }

   if abs(left_tree,right_tree) <=  1{
       return max(left_tree,right_tree) + 1
   }

   return -1;
}

func abs(i,j int)int{
    if i >j {
        return i - j
    }

    return j - i
}

func max(i,j int)int{
    if i > j{
        return i
    }

    return j
}

func isBalanced(root *TreeNode) bool {
    //
    if root == nil{
        return true
    }
    if abs(dfs(root.Left),dfs(root.Right)) <= 1{
        return isBalanced(root.Left) && isBalanced(root.Right)
    }

    return false
}
func dfs(root *TreeNode)int{
    if root == nil{
        return 0
    }

    left_tree := dfs(root.Left)
    right_tree := dfs(root.Right)

    return max(left_tree,right_tree) + 1
}

func max(i,j int)int{
    if i > j{
        return i
    }
    return j
}

func abs(i,j int)int{
    if i > j{
        return i - j
    }
    return j - i
}