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 个结点。
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 := 0
for _, 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 *= 2
b--
}
return a / 2
}