二叉树的下一个节点(详解,折磨我两个小时的题,必须得给它办得明明白白的)(数据结构基础回顾)(剑指offer)

题目如下所示:

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

它的数据结构如下所示

/*
struct TreeLinkNode {
    int val;
    struct TreeLinkNode *left;
    struct TreeLinkNode *right;
    struct TreeLinkNode *next;
    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {
        
    }
};
*/

乍一看,很简单嘛,不就是根据中序遍历求给出的节点的下一个节点,更何况题目已经把它的父节点用next指针指出来了。只要根据中序遍历的规则来求就可以了,但是万万没想到的是,我在做这道题目时,明明考虑的很清楚,很全面了,但是还是会出现下图的情况:

这让我很抓狂,我不明白为什么会出现以下的情况,我不断的模拟运行着代码,甚至还在vs中编写了一个二叉树的程序来运行我写的代码,但是运行出来的都是正确的,这让我陷入深深的思考,我觉得语法上应该没有什么问题,所以我在想是不是有一种极为特殊的情况导致这个代码出现这种情况,终于在我试了很多种极端情况下,终于让我发现一种情况下会出现这样的状况,即

二叉树为左单支,而选择的找“下一个节点”的节点pNode为根节点,由于我没有加入判断pNode的next指针是否为空,出现的以上的错误。

抛开上述的错误而言,这道题目其实本身挺简单的,可以分为两种情况。

第一种情况是给出的pNode节点右孩子不为空,根据中序遍历规则,pNode右子树的最底层的左叶子节点便是要寻找的下一个节点,代码如图所示:

if(pNode->right!=nullptr)
        {
            TreeLinkNode *k=pNode->right;
            while(k->left!=nullptr)
            {
                k=k->left;
            }
            p=k;
        }

第二情况便是pNode节点右孩子为空,这类方法,根据中序遍历规则,往上寻找双亲节点,当pNode节点是双亲节点的左孩子时,寻找的下一个节点便是双亲,代码如下:


            TreeLinkNode *k=pNode->next;
            if(k!=nullptr&&k->left==pNode)
            {
                p=k;
            }

 

而当pNode节点是双亲节点的右孩子时,就要循环去找双亲节点,当循环找到pNode节点是双亲的左孩子为止。代码如下:

TreeLinkNode * q=pNode;
                  while(k != nullptr  && q != k->left)
                  {
                      q=k;
                      k=k->next;
                  }
                p=k;

全文代码如图所示:

/*
struct TreeLinkNode {
    int val;
    struct TreeLinkNode *left;
    struct TreeLinkNode *right;
    struct TreeLinkNode *next;
    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {
        
    }
};
*/
class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        if(pNode==nullptr)
        {
            return nullptr;
        }
        TreeLinkNode *p=nullptr;
        if(pNode->right!=nullptr)
        {
            TreeLinkNode *k=pNode->right;
            while(k->left!=nullptr)
            {
                k=k->left;
            }
            p=k;
        }
        else
        {
            TreeLinkNode *k=pNode->next;
            if(k!=nullptr&&k->left==pNode)
            {
                p=k;
            }
            else
            {
                TreeLinkNode * q=pNode;
                  while(k != nullptr  && q != k->left)
                  {
                      q=k;
                      k=k->next;
                  }
                p=k;
            }
        }
        return p;
    }
};

希望对你所有帮助!!!

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

三少爷的剑!

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值