二叉树及常用算法

参考:二叉树知识点最详细最全讲解

百度
跳转

种类:
1. 满二叉树
2. 完全二叉树
3. 二叉搜索树
4. 平衡二叉树搜索树:又被称为AVL(Adelson-Velsky and Landis)树,且具有以下性质:它是一颗空树或者它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一颗平衡二叉树

存储方式:
1.链式存储
2.顺序存储:父节点的数组下标是i,那么它的左孩子就是2 * i + 1,右孩子就是i * 2 + 2

遍历方式:
1.深度优先遍历:前序遍历、中序遍历、后续遍历
2.广度优先遍历

力扣题目

  1. 二叉树的前序遍历
  2. 二叉树的后序遍历
  3. 二叉树的中序遍历
  4. 二叉树的层序遍历
  5. 二叉树的层序遍历 II
  6. 二叉树的右视图
  7. 二叉树的层平均值
  8. N 叉树的层序遍历
  9. 在每个树行中找最大值
  10. 填充每个节点的下一个右侧节点指针
  11. 填充每个节点的下一个右侧节点指针 II
  12. 二叉树的最大深度
  13. 二叉树的最小深度
  14. 翻转二叉树
  15. N 叉树的前序遍历
  16. N 叉树的后序遍历
  17. 对称二叉树


    力扣二叉树经典面试题1

    力扣二叉树经典面试题2

二叉树链式存储结构

class TreeNode {  
    var val: Int  
    var left: TreeNode?  
    var right: TreeNode?  
}

1. 二叉树的前序、中序、后续遍历

前序遍历

func preorderTraversal(_ root: TreeNode?) -> [Int] {  
    var result: [Int] = []  
    guard let root = root else { return result } // 如果根节点为空,直接返回空数组  
      
    // 递归实现前序遍历  
    preorderTraversalHelper(root, &result)  
      
    return result  
}  
  
func preorderTraversalHelper(_ node: TreeNode, _ result: inout [Int]) {  
    if let node = node {  
        // 访问根节点  
        result.append(node.val)  
        // 遍历左子树  
        preorderTraversalHelper(node.left, &result)  
        // 遍历右子树  
        preorderTraversalHelper(node.right, &result)  
    }  
}

//中序遍历

func inorderTraversal(_ root: TreeNode?) -> [Int] {  
    var result: [Int] = []  
    guard let root = root else { return result } // 如果根节点为空,直接返回空数组  
      
    // 递归实现中序遍历  
    inorderTraversalHelper(root, &result)  
      
    return result  
}  
  
func inorderTraversalHelper(_ node: TreeNode, _ result: inout [Int]) {  
    if let node = node {  
        // 遍历左子树  
        inorderTraversalHelper(node.left, &result)  
        // 访问根节点  
        result.append(node.val)  
        // 遍历右子树  
        inorderTraversalHelper(node.right, &result)  
    }  
}

后续遍历

func postorderTraversal(_ root: TreeNode?) -> [Int] {  
    var result: [Int] = []  
    guard let root = root else { return result } // 如果根节点为空,直接返回空数组  
      
    // 递归实现后序遍历  
    postorderTraversalHelper(root, &result)  
      
    return result  
}  
  
func postorderTraversalHelper(_ node: TreeNode, _ result: inout [Int]) {  
    if let node = node {  
        // 遍历左子树  
        postorderTraversalHelper(node.left, &result)  
        // 遍历右子树  
        postorderTraversalHelper(node.right, &result)  
        // 访问根节点  
        result.append(node.val)  
    }  
}

2. 二叉树的高度、节点数、最大深度、最小深度

高度 = 最大深度

func height(_ root: TreeNode?) -> Int {  
    guard let root = root else { return 0 } // 如果根节点为空,返回 0  
      
    // 递归计算左子树和右子树的高度,然后取较大值加 1  
    let leftHeight = height(root.left)  
    let rightHeight = height(root.right)  
    return max(leftHeight, rightHeight) + 1  
}

func maxDepth(_ root: TreeNode?) -> Int {  
    guard let root = root else { return 0 } // 如果根节点为空,返回 0  
      
    // 递归计算左子树和右子树的最大深度  
    let leftDepth = maxDepth(root.left)  
    let rightDepth = maxDepth(root.right)  
      
    // 返回左右子树中较大的深度加 1(根节点本身)  
    return max(leftDepth, rightDepth) + 1  
}

节点数

func countNodes(_ root: TreeNode?) -> Int {  
    guard let root = root else { return 0 } // 如果根节点为空,返回 0  
      
    // 递归计算左子树和右子树的节点数,然后相加并加上根节点本身  
    let leftCount = countNodes(root.left)  
    let rightCount = countNodes(root.right)  
    return leftCount + rightCount + 1  
}

最小深度

func minDepth(_ root: TreeNode?) -> Int {  
    guard let root = root else { return 0 } // 如果根节点为空,返回 0  
      
    // 如果左子树为空且右子树不为空,返回右子树的最小深度加 1  
    if root.left == nil && root.right != nil {  
        return minDepth(root.right) + 1  
    }  
      
    // 如果右子树为空且左子树不为空,返回左子树的最小深度加 1  
    if root.right == nil && root.left != nil {  
        return minDepth(root.left) + 1  
    }  
      
    // 如果左右子树都不为空,返回左右子树最小深度的较小值加 1  
    if let leftDepth = root.left, let rightDepth = root.right {  
        return min(minDepth(leftDepth), minDepth(rightDepth)) + 1  
    }  
      
    // 如果左右子树都为空,说明当前节点是叶子节点,返回 1  
    return 1  
}  

3.二叉树的层序遍历、右视图、层平均值、是否为完全二叉树

层序遍历

func levelOrder(_ root: TreeNode?) -> [[Int]] {  
    guard let root = root else { return [] } // 如果根节点为空,直接返回空数组  
      
    var result: [[Int]] = [] // 用于存储每一层的节点值  
    var queue: [TreeNode] = [root] // 使用队列进行层序遍历  
      
    while !queue.isEmpty {  
        let levelSize = queue.count // 当前层的节点数  
        var level: [Int] = [] // 存储当前层的节点值  
          
        for _ in 0..<levelSize {  
            let node = queue.removeFirst() // 取出队首节点  
            level.append(node.val) // 将节点值加入当前层数组  
              
            // 将当前节点的左、右子节点(如果存在)入队  
            if let left = node.left {  
                queue.append(left)  
            }  
            if let right = node.right {  
                queue.append(right)  
            }  
        }  
          
        result.append(level) // 将当前层数组加入结果数组  
    }  
      
    return result // 返回按层遍历的结果  
}  

右视图

func rightSideView(_ root: TreeNode?) -> [Int] {  
    guard let root = root else { return [] } // 如果根节点为空,返回空数组  
      
    var result = [Int]() // 存储每一层最右侧节点的值的数组  
    var queue = [TreeNode]() // 用于层序遍历的队列  
      
    queue.append(root) // 将根节点加入队列  
      
    while !queue.isEmpty {  
        let levelSize = queue.count // 当前层的节点数量  
          
        // 遍历当前层的所有节点  
        for _ in 0..<levelSize {  
            let node = queue.removeFirst() // 取出并移除队列中的第一个节点  
              
            // 将当前节点最右侧的值添加到结果数组中  
            if queue.isEmpty || (queue.first?.left == nil && queue.first?.right == nil) { // 判断是否为本层最右侧节点  
                result.append(node.val)  
            }  
              
            // 将当前节点的子节点加入队列  
            if let left = node.left {  
                queue.append(left)  
            }  
            if let right = node.right {  
                queue.append(right)  
            }  
        }  
    }  
      
    return result  
}  

层平均值
要计算二叉树每一层的平均值(即层平均值),你仍然需要执行层序遍历(广度优先遍历),但这次在遍历每一层时,你需要累加该层所有节点的值,并除以该层的节点数来得到平均值。

func averageOfLevels(_ root: TreeNode?) -> [Double] {  
    guard let root = root else { return [] } // 如果根节点为空,返回空数组  
      
    var result = [Double]() // 存储每一层平均值的数组  
    var queue = [TreeNode]() // 用于层序遍历的队列  
      
    queue.append(root) // 将根节点加入队列  
      
    while !queue.isEmpty {  
        let levelSize = queue.count // 当前层的节点数量  
        var levelSum = 0 // 当前层的节点值之和  
          
        // 遍历当前层的所有节点  
        for _ in 0..<levelSize {  
            let node = queue.removeFirst() // 取出并移除队列中的第一个节点  
              
            levelSum += node.val // 累加当前节点的值  
              
            // 将当前节点的子节点加入队列  
            if let left = node.left {  
                queue.append(left)  
            }  
            if let right = node.right {  
                queue.append(right)  
            }  
        }  
          
        // 计算当前层的平均值并添加到结果数组中  
        let average = Double(levelSum) / Double(levelSize)  
        result.append(average)  
    }  
      
    return result  
} 

完全二叉树
在二叉树中,完全二叉树是指除了最后一层外,每一层上的节点数都达到最大值,并且最后一层上的节点都靠左紧密排列的二叉树。要判断一个二叉树是否是完全二叉树,我们可以使用层序遍历(广度优先搜索)的方法,并利用一个标记位来记录是否遇到了空节点。
一旦我们遇到了一个空节点,我们就应该确保后续的节点都是空的,因为完全二叉树的特点是:如果某个节点为空,那么它后面所有的节点都应为空。

func isCompleteTree(_ root: TreeNode?) -> Bool {  
    guard let root = root else { return true } // 空树视为完全二叉树  
      
    var queue = [TreeNode]()  
    var encounteredNull = false // 标记是否遇到了空节点  
      
    queue.append(root)  
      
    while !queue.isEmpty {  
        let node = queue.removeFirst()  
          
        if node.left != nil {  
            if encounteredNull {  
                // 如果之前已经遇到了空节点,但当前节点还有左子节点,则不是完全二叉树  
                return false  
            }  
            queue.append(node.left!)  
        } else {  
            // 遇到了空节点,将 encounteredNull 标记为 true  
            encounteredNull = true  
        }  
          
        if node.right != nil {  
            if encounteredNull {  
                // 如果之前已经遇到了空节点,但当前节点还有右子节点,则不是完全二叉树  
                return false  
            }  
            queue.append(node.right!)  
        } else {  
            // 如果右子节点为空,并且之前没遇到过空节点,则 encounteredNull 依然为 false  
            // 无需额外操作,继续遍历  
        }  
    }  
      
    return true // 遍历完成,未违反完全二叉树的定义,则返回 true  
} 

满二叉树
满二叉树(Full Binary Tree)是一种特殊的二叉树,其中除了叶子节点外,每个节点都有两个子节点。换句话说,满二叉树的每一层都完全填满,并且所有叶子节点都在同一层。

要判断一个二叉树是否为满二叉树,我们可以使用递归的方法。在递归过程中,我们需要确保每个非叶子节点都有两个子节点,并且所有叶子节点都位于同一层(这通常通过遍历过程中记录的层数来判断)。然而,在实际应用中,我们往往不需要显式地检查叶子节点是否在同一层,因为如果我们能确保每个非叶子节点都有两个子节点,那么叶子节点自然会在同一层。

func isFullBinaryTree(_ root: TreeNode?) -> Bool {  
    guard let root = root else { return true } // 空树视为满二叉树  
      
    // 递归检查左右子树是否都是满二叉树,并且确保当前节点有两个子节点  
    return (root.left != nil && root.right != nil) && // 当前节点有两个子节点  
           isFullBinaryTree(root.left) && // 左子树是满二叉树  
           isFullBinaryTree(root.right) // 右子树是满二叉树  
}  

4.对称二叉树

要判断一个二叉树是否是对称的,可以采用递归的方法。对称二叉树意味着二叉树的左子树与右子树在结构上镜像对称,且每个节点的值都相等。

func isSymmetric(_ root: TreeNode?) -> Bool {  
    guard let root = root else { return true } // 空树是对称的  
    return isMirror(root.left, root.right)  
}  
  
// 辅助函数,判断两棵树是否镜像对称  
func isMirror(_ t1: TreeNode?, _ t2: TreeNode?) -> Bool {  
    if t1 == nil && t2 == nil {  
        return true  
    }  
    if t1 == nil || t2 == nil {  
        return false  
    }  
    return (t1?.val == t2?.val) &&  
        isMirror(t1?.left, t2?.right) && // 注意这里是左右互换  
        isMirror(t1?.right, t2?.left)  
}  
  
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容