面試題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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章