題目要求:
給定一棵二叉樹和一個節點,找出這個節點在中序序列中的下一個節點。
自己的思路:
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;
}