958. 二叉树的完全性检验 golang

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]bool
                                                                                                                 
func 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] = true

    id << 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 := 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
}