题目描述:
给定一个二叉树其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的next指针。下图为一棵有9个节点的二叉树。树中从父节点指向子节点的指针用实线表示,从子节点指向父节点的用虚线表示:
示例:
输入:{8,6,10,5,7,9,11},8
返回:9
解析:这个组装传入的子树根节点,其实就是整颗树,中序遍历{5,6,7,8,9,10,11},根节点8的下一个节点就是9,应该返回{9,10,11},后台只打印子树的下一个节点,所以只会打印9,如下图,其实都有指向左右孩子的指针,还有指向父节点的指针,下图没有画出来:
数据范围:
节点数满足 1≤n≤50, 节点上的值满足 1≤val≤100
要求:
空间复杂度 O(1),时间复杂度 O(n)
输入描述:
输入分为2段,第一段是整体的二叉树,第二段是给定二叉树节点的值,后台会将这2个参数组装为一个二叉树局部的子树传入到函数GetNext里面,用户得到的输入只有一个子树根节点
返回值描述:
返回传入的子树根节点的下一个节点,后台会打印输出这个节点
示例:
输入:{8,6,10,5,7,9,11},8
返回值:9
输入:{8,6,10,5,7,9,11},6
返回值:7
输入:{1,2,#,#,3,#,4},4
返回值:1
输入:{5},5
返回值:“null”
实现代码
方法一:二叉树中序遍历
TreeLinkNode* GetNext(TreeLinkNode* pNode) {
if(pNode == nullptr)
return nullptr;
int val = pNode->val;
while(pNode->next) //寻找根节点
pNode = pNode->next;
stack<TreeLinkNode*> sk;
int flag = 1;
while(!sk.empty() || pNode != nullptr){
while(pNode != nullptr){
sk.push(pNode);
pNode = pNode->left;
}
if(!sk.empty()){
pNode = sk.top();
sk.pop();
if(flag == 0)
return pNode;
if(pNode->val == val)
flag = 0;
pNode = pNode->right;
}
}
return nullptr;
}
方法二:利用中序遍历性质
中序遍历的步骤可表示为LTR,其中L代表左子树,T代表根结点,R代表右子树,即:先访问某结点的左子树,再访问该结点,再访问该结点的右子树。
「某结点的下一个结点」仅有二种情况:
- 若该结点存在右子树,则「下一个结点」为「该结点右子树的最左孩子」。
- 若该结点不存在右子树,则「下一个结点」为「该结点的第一个右父结点」。
TreeLinkNode* GetNext(TreeLinkNode* pNode) {
if (!pNode)
return NULL;
// 若该结点存在右子树
if (pNode->right) {
TreeLinkNode* rightChild = pNode->right;
// 寻找右子树上的最左孩子
while (rightChild->left)
rightChild = rightChild->left;
return rightChild;
}
// 若不存在右子树,寻找第一个右父亲
while (pNode->next) {
if (pNode->next->left == pNode)
return pNode->next;
pNode = pNode->next;
}
return NULL;
}