综合门户网站开发网络项目发布网
-
二叉树我最近刷的特别多，差不多快刷完了，但是有一种题型差点给我忽略了，那就是完全二叉树，这也是一个很重要的题型，今天刚好有一道题目可以来复习一下完全二叉树的特性
-
题目链接如下：https://leetcode.cn/problems/count-complete-tree-nodes/?envType=study-plan-v2&envId=top-interview-150
-
做这道题首先有一点要知道的，就是完全二叉树是怎么样子的，下面说一下完全二叉树的概念
-
完全二叉树：只有最底层的节点未被填满，且最底层节点尽量靠左填充
-
ok 现在已经了解了基础概念了，我们再来看这道题目
-
这道题目的目的是让我们遍历这棵树，并算有几个节点
-
说实话，这道题很简单，用暴力的做法，就是遍历一整棵树，代码如下：
-
递归法：
//方法一：递归
func Solution222(root *TreeNode) int {if root == nil {return 0}left := Solution222(root.Left)right := Solution222(root.Right)return left + right + 1
}
- 迭代法：
//方法二：迭代
func Solution222v2(root *TreeNode) int {if root == nil{return 0}queue := []*TreeNode{root}count := 0for len(queue) > 0{node := queue[0]queue = queue[1:]count++if node.Left != nil{queue = append(queue, node.Left)}if node.Right != nil{queue = append(queue, node.Right)}}return count
}
-
这两个方法是遍历树的最基本的方法之一
-
但是 这不是这道题的本意，这道题目是想要我们理由完全二叉树这个特性解题
-
那我们需要好好思考一下，完全二叉树有什么特点
-
除了最后的叶子节点，其他层级节点都是满的
-
当 左子树的深度 和 右子树的深度 一致的时候，说明 左子树是满的二叉树 可以通过 2的h次方求的左子树的节点个数
-
当 右子树的深度 不如 左子树的深度 的时候，说明 左子树不是一个满的二叉树，但是右子树单独看是一个满的二叉树，所以可以通过 2的h次方求右子树的节点个数
-
-
ok，知道这些特点，我们是不是可以利用一个逻辑来减少遍历
- 当 左子树深度 等于 右子树的时候，就可以通过深度来计算左子树的节点，然后只遍历右子树
- 当 左子树深度 等于 右子树的时候，就可以通过深度来计算右子树的节点，然后只遍历左子树
-
这样我们本来需要遍历全部二叉树节点的，现在只需要遍历一半，思路瞬间打开，代码如下：
//方法三：二分查找
func Solution222v3(root *TreeNode) int {if root == nil{return 0}//检索左子树深度left := root.Leftldepth := 0for left != nil{left = left.Leftldepth++}//检索右子树深度right := root.Rightrdepth := 0for right != nil{right = right.Leftrdepth++}//左右子树深度判断if ldepth == rdepth{return (1<<ldepth) + Solution222v3(root.Right)}else{return (1<<rdepth) + Solution222v3(root.Left)}
}
ok，这里这道题目就结束了，感谢大家观看