查找二叉樹的節點(某一值的節點)

#include <iostream>
#include <stdio.h>
#include <stdlib.h>

struct BinaryTreeNode{
    char value;
    struct BinaryTreeNode *lchild, *rchild;
};

// 輸入值依次是:a、b、#、#、c、#、#
struct BinaryTreeNode *creatBiTree() {
    struct BinaryTreeNode *node;
    char data;
    std::cin>>data;
    if (data == '#') {
        node = NULL;
    } else {
        node = (struct BinaryTreeNode *)malloc(sizeof(struct BinaryTreeNode *));
        node->value = data;
        node->lchild = creatBiTree();
        node->rchild = creatBiTree();
    };
    return node;
}

struct BinaryTreeNode * searchNode(struct BinaryTreeNode *node) {
    struct BinaryTreeNode * endNode = NULL;
    if (node == NULL) {
        return NULL;
    }
    if (node->value == 'c') {
        endNode = node;
    } else {
        if (endNode == NULL) {
            endNode =  searchNode(node->lchild);
        }
        if (endNode == NULL) {
            endNode = searchNode(node->rchild);
        }
    }
    return endNode;
}


int main(int argc, const char * argv[]) {
    struct BinaryTreeNode *rootNode = creatBiTree();
    struct BinaryTreeNode *endNode = searchNode(rootNode);
    
    printf("%d", endNode->value);
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章