牛客：在二叉树中找到一个节点的后续节点

最新推荐文章于 2025-03-18 14:35:38 发布

原创最新推荐文章于 2025-03-18 14:35:38 发布 · 533 阅读

0 ·

CC 4.0 BY-SA版权

|--二叉树同时被 2 个专栏收录

11 篇文章

订阅专栏

牛客

5 篇文章

订阅专栏

本文介绍了一种算法，用于找出给定二叉树中特定节点的后续节点，即中序遍历序列中的下一个节点。通过构建二叉树并进行中序遍历，当遍历到目标节点时，记录下一次遍历的节点作为所求的后续节点。

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;
}