一些二叉树相关算法Python的解决

本文介绍了一种使用广度优先搜索算法来确定二叉树最小深度的方法,并通过示例代码展示了如何实现这一功能。此外,还探讨了二叉树的锯齿形层序遍历,包括广度优先及深度优先两种实现方式。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

二叉树的最小深度

广度优先

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def minDepth(self, root):
       if not root:
            return 0

        queue, level = [root], 1
        while queue:
            for i in xrange(len(queue)):
                node = queue.pop(0)

                if not node.left and not node.right:
                    return level

                if node.left:
                    queue.append(node.left)

                if node.right:
                    queue.append(node.right)

            level += 1 

二叉树遍历用列表输入每层节点

广度有优先

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def zigzagLevelOrder(self, root):
        res, queue = [], [(root, 0)]
        while queue:
            curr, level = queue.pop(0)
            if curr:
                if len(res) < level+1:
                    res.append([])
                if level % 2 == 0:
                    res[level].append(curr.val)
                else:
                    res[level].insert(0, curr.val)
                queue.append((curr.left, level+1))
                queue.append((curr.right, level+1))
        return res  

深度优先


def zigzagLevelOrder(self, root):
    ans = []
    self.dfs(root, 0, ans)

    for i in range(len(ans)):
        if i%2:
            ans[i] = ans[i][::-1] # reverse

    return ans

def dfs(self, root, level, ans):
    if not root:
        return

    if len(ans) < level+1: #should >= level+1
        ans.append([])

    ans[level].append(root.val)
    self.dfs(root.left, level+1, ans)
    self.dfs(root.right, level+1, ans)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值