【PTA数据结构 | C语言版】根据前序序列重构二叉树

本专栏持续输出数据结构题目集,欢迎订阅。

文章目录

题目

请编写程序,根据给定二叉树的前序序列化结果,重构二叉树,并输出其前序遍历结果。

输入格式:

输入首先给出一个不超过 20 的正整数 n,随后一行给出 n 个前序序列的元素。其中键值都是不超过 9 位的正整数,空结点对应符号 #。

输出格式:

输出二叉树的前序遍历结果,每个数字占一行。

输入样例:

11

1 2 # 4 # # 3 5 # # #

输出样例:

1

2

4

3

5

代码

cpp 复制代码
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

typedef struct TreeNode {
    int data;
    struct TreeNode *left;
    struct TreeNode *right;
} TreeNode;

// 创建新节点
TreeNode* createNode(int data) {
    TreeNode* node = (TreeNode*)malloc(sizeof(TreeNode));
    node->data = data;
    node->left = NULL;
    node->right = NULL;
    return node;
}

// 根据前序序列化结果递归构建二叉树
TreeNode* buildTree(char** tokens, int* index, int n) {
    if (*index >= n || strcmp(tokens[*index], "#") == 0) {
        (*index)++;
        return NULL;
    }
    
    TreeNode* node = createNode(atoi(tokens[*index]));
    (*index)++;
    
    node->left = buildTree(tokens, index, n);
    node->right = buildTree(tokens, index, n);
    
    return node;
}

// 前序遍历二叉树(根-左-右)
void preOrderTraversal(TreeNode* root) {
    if (root == NULL) return;
    
    printf("%d\n", root->data);
    preOrderTraversal(root->left);
    preOrderTraversal(root->right);
}

int main() {
    int n;
    scanf("%d", &n);
    
    // 消耗掉scanf后的换行符
    getchar();
    
    char input[1000];
    fgets(input, sizeof(input), stdin);
    
    // 分割输入字符串为标记数组
    char* tokens[100];
    int count = 0;
    char* token = strtok(input, " \n");
    
    while (token != NULL && count < n) {
        tokens[count++] = token;
        token = strtok(NULL, " \n");
    }
    
    int index = 0;
    TreeNode* root = buildTree(tokens, &index, n);
    
    preOrderTraversal(root);
    
    return 0;
}
相关推荐
我搞slam4 小时前
快乐数--leetcode
算法·leetcode·哈希算法
WWZZ20254 小时前
快速上手大模型:机器学习3(多元线性回归及梯度、向量化、正规方程)
人工智能·算法·机器学习·机器人·slam·具身感知
东方佑5 小时前
从字符串中提取重复子串的Python算法解析
windows·python·算法
西阳未落6 小时前
LeetCode——二分(进阶)
算法·leetcode·职场和发展
通信小呆呆6 小时前
以矩阵视角统一理解:外积、Kronecker 积与 Khatri–Rao 积(含MATLAB可视化)
线性代数·算法·matlab·矩阵·信号处理
Dfreedom.6 小时前
一文掌握Python四大核心数据结构:变量、结构体、类与枚举
开发语言·数据结构·python·变量·数据类型
CoderCodingNo7 小时前
【GESP】C++四级真题 luogu-B4068 [GESP202412 四级] Recamán
开发语言·c++·算法
一个不知名程序员www7 小时前
算法学习入门---双指针(C++)
c++·算法
Want5957 小时前
C/C++大雪纷飞①
c语言·开发语言·c++
Shilong Wang8 小时前
MLE, MAP, Full Bayes
人工智能·算法·机器学习