题解 | #二叉树的下一个结点#
二叉树的下一个结点
http://www.nowcoder.com/practice/9023a0c988684a53960365b889ceaf5e
C语言求二叉树的下一个中序遍历节点
- 思路:注意中序遍历,那么就去写一个中序遍历的递归函数;注意每个节点的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;
# }