【牛客刷题专栏】0x10:JZ8 二叉树的下一个结点(C语言编程题)

前言

• 个人推荐在牛客网刷题(点击可以跳转)，它登陆后会保存刷题记录进度，重新登录时写过的题目代码不会丢失。
• 个人刷题练习系列专栏：个人CSDN牛客刷题专栏。 题目来自：牛客/题库 / 在线编程 / 剑指offer：
  

---

问题描述：

• 给定一个二叉树其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。
• 注意，树中的结点不仅包含左右子结点，同时包含指向父结点的next指针。
• 下图为一棵有9个节点的二叉树。树中从父节点指向子节点的指针用实线表示，从子节点指向父节点的用虚线表示.
  

//示例1：
//输入：
{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(n)，时间复杂度 O(n)

• 输入描述：
  输入分为2段，第一段是整体的二叉树，第二段是给定二叉树节点的值，后台会将这2个参数组装为一个二叉树局部的子树传入到函数GetNext里面，用户得到的输入只有一个子树根节点

• 返回值描述：
  返回传入的子树根节点的下一个节点，后台会打印输出这个节点

---

举例：

//示例1：
//输入：
{8,6,10,5,7,9,11},8
//返回值：
9
//=================
//示例2：
//输入：
{8,6,10,5,7,9,11},6
//返回值：
7
//=================
//示例3：
//输入：
{1,2,#,#,3,#,4},4
//返回值：
1
//=================
//示例3：
//输入：
{5},5
//返回值：
"null"
//说明：不存在，后台打印"null"   

---

解法思路：

• 思路：注意中序遍历，那么就去写一个中序遍历的递归函数；注意每个节点的Next指针指向父节点，说明尽头是根节点。这样就可以去中序遍历了；
• 重点：使用了一个全局静态变量flag，用于标志是否中序遍历到目标节点，如果遍历到目标节点，说明下一个节点就是我们需要返回的节点。可以根据flag来判断。

---

代码结果：

 /* struct TreeLinkNode {
 *	int val;
 *	struct TreeLinkNode *left;
 *	struct TreeLinkNode *right;
 *	struct TreeLinkNode *next;
 * };
 *
 * C语言声明定义全局变量请加上static，防止重复定义
 */
/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param pNode TreeNode类 
 * @return TreeNode类
 */
//target存放寻找的节点
//returnNode 保存目标的下一个节点
static struct TreeLinkNode *target=NULL;
static struct TreeLinkNode *returnNode=NULL;
//flag用来标志是否寻找到target
static int flag=0;
//一个中序遍历的递归函数
int MidSearch(struct TreeLinkNode * pRoot){
    if(pRoot==NULL)
        return 0;
    MidSearch(pRoot->left);
    //先判断flag是否为1，判断是否匹配到上一个中序节点
    //flag用完重置
    if(flag==1){
        returnNode=pRoot;
        flag=0;
    }
     //表示找到了目标节点，设置flag,返回下一次中序遍历结果   
    if(target==pRoot)
        flag=1;
    MidSearch(pRoot->right);
    return 0;
}
struct TreeLinkNode* GetNext(struct TreeLinkNode* pNode ) {
    // write code here、
    target=pNode;
    //第一步找到根节点
    while(pNode->next)
        pNode=pNode->next;
    MidSearch(pNode);
    //返回全局变量
    return returnNode;
}

---

结束语

• 以上就是该C语言编程题的内容。可以在牛客尝试刷几道题目来练习实践。牛客网刷题(点击可以跳转)，可以尝试注册使用。
• 题目来自：牛客/题库 / 在线编程 / 剑指offer：