python剑指offer系列二叉树的下一个节点

本文介绍了一种算法,用于确定给定二叉树中任一节点的中序遍历序列中的下一个节点。该算法首先找到指定节点的根节点,并通过中序遍历将所有节点保存到数组中,最后返回目标节点后的下一个节点。

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

题目:

给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。



solution


思路:

第一步还是找到某个节点的根节点,方法是一直使用next判断即可。

再将从根节点中序遍历的结果保存到一个数组中,直接找pNode所在索引的下一个即可。当然要考虑这个节点是不是最后一个,如果是最后一个,直接返回None。

代码:

# -*- coding:utf-8 -*-
# class TreeLinkNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#         self.next = None
class Solution:
    def GetNext(self, pNode):
        if not pNode:
            return 
        pRoot = pNode
        tmp = pNode
        while pNode:
            if pNode.next == None:
                break
            pRoot = pNode.next
            pNode = pNode.next
        self.arr = []
        self.midTrave(pRoot)
        for i in range(len(self.arr)):
            if self.arr[i] == tmp:
                if i <=len(self.arr) - 2:
                    return self.arr[i+1]
                else:
                    return 
                     
    def midTrave(self,pRoot):
        if not pRoot :return 
        self.midTrave(pRoot.left)
        self.arr.append(pRoot)
        self.midTrave(pRoot.right)
        # write code here

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值