面试题8——寻找二叉树的下一个节点

题目要求:
给定一棵二叉树和一个节点,找出这个节点在中序序列中的下一个节点。

自己的思路:
1、可以根据中序遍历构建一个中序线索二叉树,然后所有结点的中序遍历的前一个节点和后一个节点都可以找出来了。
2、根据中序遍历的特点,找出所有的可能性,既可以判断中序遍历的下一个节点。在这里插入图片描述
实现代码:

#include<iostream>
#include<string>
#include<stack>

using namespace std;
struct TreeNode{
    int data;
    TreeNode *leftchild;
    TreeNode *rightchild;
    TreeNode  *parent;
    int left_tag;
    int right_tag;
}
//build thread_tree
TreeNode *ThreadTree(TreeNode *T,TreeNode *pre){
    if(T){
        ThreadTree(T->leftchild, pre);
        if(T->leftchild==nullptr){
            T->leftchild = pre;
            T->left_tag = 1;
        }
        if(pre->rightchild == nullptr){
            pre->rightchild = T;
            pre->right_tag = 1;
        }
        pre = T;
        ThreadTree(T->rightchild, pre);
    }
}
int main(TreeNode *T){
    
    TreeNode *pre = new TreeNode;
    pre = nullptr;
    if(T){
        ThreadTree(T);
    }
    pre->rightchild = nullptr;
    pre->right_tag = 0;
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章