958. 二叉树的完全性检验
给定一个二叉树,确定它是否是一个完全二叉树。
百度百科中对完全二叉树的定义如下:
若设二叉树的深度为 h,除第 h 层外,其它各层 (1~h-1) 的结点数都达到最大个数,第 h 层所有的结点都连续集中在最左边,这就是完全二叉树。(注:第 h 层可能包含 1~ 2h 个节点。)
示例 1:
输入:[1,2,3,4,5,6]
输出:true
解释:最后一层前的每一层都是满的(即,结点值为 {1} 和 {2,3} 的两层),且最后一层中的所有结点({4,5,6})都尽可能地向左。
示例 2:
输入:[1,2,3,4,5,null,7]
输出:false
解释:值为 7 的结点没有尽可能靠向左侧。
提示:
树中将会有 1 到 100 个结点。
解决
var ids map[int]boolfunc isCompleteTree(root *TreeNode) bool {ids = make(map[int]bool)dfs(root, 0)for i:=0; i<len(ids); i++ {if !ids[i] {return false} } return true
}func dfs(node *TreeNode, id int) {if node == nil {return } ids[id] = trueid << 1 dfs(node.Left, id + 1)dfs(node.Right, id + 2)
}
Code
var res [][]int
// 入口
func isCompleteTree(root *TreeNode) bool {res = [][]int{}dfs(root, 0)//判断前几个结点是否饱和for i:=0; i<=len(res)-2; i++ {m := pow(2, i)if len(res[i]) != m {return false}}//如果出现了空节点,那么后面如果还有数字,那么就肯定不是满二叉树m := 0for _, v := range res[len(res)-2] {if v == 999 {m = 1}if m == 1 && v != 999 {return false}}return true
}// 层序遍历代码
func dfs(root *TreeNode, level int) {if len(res) == level {res = append(res, []int{})}if root != nil {res[level] = append(res[level], root.Val)dfs(root.Left, level+1)dfs(root.Right, level+1)} else {res[level] = append(res[level], 999)}
}
// 求整数幂函数
func pow(a, b int) int {if b == 0 {return 1}for b != 0 {a *= 2b--}return a / 2
}