五分钟掌握:二叉树的下一个节点
Posted 一个搬砖的胖子
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了五分钟掌握:二叉树的下一个节点相关的知识,希望对你有一定的参考价值。
前言
今天补充的题目虽是剑指offer的题,但Leetcode上没有。
在各大公司面试都容易被问,加在补充题也是希望能引起大家的注意。
一起花5min学习一下这道题吧。
题目描述
给定二叉树其中的一个结点,请找出其中序遍历顺序的下一个结点并且返回。
注意,树中的结点不仅包含左右子结点,而且包含指向父结点的指针。
题目分析
以下解析参考牛客@刘豪杰,思路非常清晰。
节点(设为x)中序遍历的下一个节点有以下可能:
-
若x有右子树。则x的下一个节点为x右子树最左侧节点。如,2的下一个节点为8。 -
若x没有右子树,又分为2种情况。
-
若x是父节点的左孩子。则x的父节点就是x的下一个节点。如,7的下一个节点是4。 -
若x是父节点的右孩子。则沿着父节点向上,直到找到一个节点的父节点的左孩子是该节点,则该节点的父节点就是x的下一个节点。如,9的下一个节点是1。
我个人觉得这题并不简单。之前我也做过这道题,但是这两天去做,又遗漏了一种情况。看来重复刷题真的很重要!!
掌握了的话,提交代码试试~
参考代码
//参考牛客@weizier的JAVA代码
/*
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->right) { //如果有右子树,则找右子树的最左节点
TreeLinkNode *p = pNode->right;
while(p->left) p = p->left;
return p;
}
TreeLinkNode *p = pNode;
while(p->next) { //没右子树,则找第一个当前节点是父节点左孩子的节点
if(p->next->left == p) return p->next;
p = p->next;
}
return nullptr; //退到了根节点仍没找到,则返回null
}
};
以上是关于五分钟掌握:二叉树的下一个节点的主要内容,如果未能解决你的问题,请参考以下文章