题目要求:
给定一棵二叉树和一个节点,找出这个节点在中序序列中的下一个节点。
自己的思路:
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;
}