【PTA数据结构 | C语言版】二叉树层序序列化

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

题目

请编写程序，创建一棵有 3 个结点的二叉树，并输出其层序序列化结果。

输入格式：
输入给出 3 个整数，依次为二叉树根结点的左孩子、右孩子、根结点本身存储的键值。

输出格式：
输出二叉树的层序序列化结果，每个数字占一行，空结点对应符号 #。

输入样例：
1 2 3

输出样例：
3
1
2
#
#
#
#

代码

#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* createTree(int left_val, int right_val, int root_val) {
    TreeNode* root = createNode(root_val);
    root->left = createNode(left_val);
    root->right = createNode(right_val);
    return root;
}

// 层序遍历序列化
void levelOrderSerialize(TreeNode* root) {
    if (root == NULL) return;
    
    // 定义队列
    TreeNode* queue[1000];
    int front = 0, rear = 0;
    
    queue[rear++] = root;
    
    while (front < rear) {
        TreeNode* current = queue[front++];
        
        if (current == NULL) {
            printf("#\n");
            continue;
        }
        
        printf("%d\n", current->data);
        
        // 无论子节点是否为空都入队
        queue[rear++] = current->left;
        queue[rear++] = current->right;
    }
}

int main() {
    int left_val, right_val, root_val;
    scanf("%d %d %d", &left_val, &right_val, &root_val);
    
    TreeNode* root = createTree(left_val, right_val, root_val);
    levelOrderSerialize(root);
    
    return 0;
}