https://www.nowcoder.com/questionTerminal/c37ec6a9e4084b9c943be2d3a369e177
题目描述:给出一棵二叉树,给出树中的一个节点,求这个节点的后续节点。二叉树中一个节点的后继节点指的是,二叉树的中序遍历的序列中的下一个节点。
思路:首先建树,然后中序遍历。当遍历到给定节点时,做记录,中序遍历的下一个节点即为所求。以下代码需要中序遍历整棵树,因为没有找到停止递归的方法,也就是跳出递归函数,直接跳到main函数。
#include<stdio.h>
#include<string.h>
#include<malloc.h>
#define N_MAX 500000
int fa_index[N_MAX+10][2], node_next=0; // node_next即为所求,记录所求值
bool judge;
struct BinTree{
int data;
BinTree *lchild, *rchild;
} *root;
BinTree* CreateTree(int fa_index[][2], int fa){
if(fa==0) return NULL;
BinTree *q;
q = (BinTree *)malloc(sizeof(BinTree));
q->data = fa;
q->lchild = CreateTree(fa_index, fa_index[fa][0]);
q->rchild = CreateTree(fa_index, fa_index[fa][1]);
return q;
}
void InOrder(BinTree* head, int node){
if(head){
InOrder(head->lchild, node);
if(judge){
//printf("%d\n", head->data);
//return; //如何跳出递归?
node_next = head->data;
judge = false;
}
if(head->data==node) judge = true;
InOrder(head->rchild, node);
}
}
void Release(BinTree* head){
if(head){
Release(head->lchild);
Release(head->rchild);
free(head);
}
}
int main(){
int n, r;
while(scanf("%d %d", &n, &r)!=EOF){
while(n--){
int fa, lch, rch;
scanf("%d %d %d", &fa, &lch, &rch);
fa_index[fa][0] = lch;
fa_index[fa][1] = rch;
}
root = CreateTree(fa_index, r);
int node;
scanf("%d", &node);
judge = false;
InOrder(root, node);
printf("%d\n", node_next);
}
return 0;
}