C语言经典算法之二叉树的后序线索化

JJJ69

于 2024-01-20 16:28:35 发布

阅读量896

点赞数 22

分类专栏： C语言经典算法文章标签：算法 c语言数据结构

本文链接：https://blog.csdn.net/weixin_56154577/article/details/135717439

版权

C语言经典算法专栏收录该内容

160 篇文章 13 订阅

订阅专栏

前言

A.建议

1.学习算法最重要的是理解算法的每一步，而不是记住算法。

2.建议读者学习算法的时候，自己手动一步一步地运行算法。

tips：文中的对数均以2为底数】

B.简介

后序线索化是将二叉树按照后序遍历的顺序进行线索化的过程。线索化的目的是为了在遍历时能够更高效地访问节点。

一代码实现

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

// 定义二叉树节点结构
struct TreeNode {
    int data;
    struct TreeNode* left;
    struct TreeNode* right;
    int isThreaded;  // 标记是否为线索
};

// 创建新的二叉树节点
struct TreeNode* createNode(int data) {
    struct TreeNode* newNode = (struct TreeNode*)malloc(sizeof(struct TreeNode));
    if (newNode == NULL) {
        perror("Memory allocation failed");
        exit(EXIT_FAILURE);
    }

    newNode->data = data;
    newNode->left = NULL;
    newNode->right = NULL;
    newNode->isThreaded = 0;

    return newNode;
}

// 后序线索化函数
void createPostorderThread(struct TreeNode* root, struct TreeNode** prev) {
    if (root == NULL) {
        return;
    }

    // 递归线索化左子树
    createPostorderThread(root->left, prev);

    // 递归线索化右子树
    createPostorderThread(root->right, prev);

    // 处理当前节点
    if (*prev != NULL) {
        if ((*prev)->right == NULL) {
            // 如果前驱节点的右指针为空，将其设置为当前节点
            (*prev)->right = root;
            (*prev)->isThreaded = 1;
        }
    }

    // 更新前驱节点
    *prev = root;
}

// 后序遍历线索化二叉树
void postorderThreadedTraversal(struct TreeNode* root) {
    struct TreeNode* current = root;

    while (current != NULL) {
        // 移动到最左子树
        while (current->left != NULL && current->isThreaded == 0) {
            current = current->left;
        }

        // 如果右子树为空或是线索化的，或者当前节点是叶子节点
        if (current->right == NULL || current->isThreaded == 1 || (current->left == NULL && current->right == NULL)) {
            printf("%d ", current->data);
            current = current->right;
        } else {
            // 否则，移动到右子树的根节点
            current = current->right;
        }
    }
}

int main() {
    // 创建示例二叉树
    struct TreeNode* root = createNode(1);
    root->left = createNode(2);
    root->right = createNode(3);
    root->left->left = createNode(4);
    root->left->right = createNode(5);

    // 执行后序线索化
    struct TreeNode* prev = NULL;
    createPostorderThread(root, &prev);

    // 执行后序遍历线索化二叉树
    printf("Postorder Threaded Traversal: ");
    postorderThreadedTraversal(root);
    printf("\n");

    return 0;
}

这个例子中，createPostorderThread 函数负责后序线索化，而 postorderThreadedTraversal 函数实现了线索化二叉树的后序遍历。后序线索化的思想类似于中序线索化，只是遍历的顺序不同。