【PTA数据结构 | C语言版】前序遍历树

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

题目

请编写程序，创建有 4 个结点的树，然后输出其前序遍历序列。

输入格式：
输入首先在第一行给出 4 个正整数，依次对应树的根结点、根的第 1、2、3 个孩子结点的键值。所有键值均为 int 型范围内的整数，同行数字间以空格分隔。

输出格式：
输出树的前序遍历序列，每个键值占一行。

输入样例：
1 2 3 4

输出样例：
1
2
3
4

代码

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

// 定义树节点结构
typedef struct TreeNode {
    int key;
    struct TreeNode* children[3];  // 最多3个子节点
} TreeNode;

// 创建新节点
TreeNode* createNode(int key) {
    TreeNode* node = (TreeNode*)malloc(sizeof(TreeNode));
    node->key = key;
    // 初始化子节点为NULL
    for (int i = 0; i < 3; i++) {
        node->children[i] = NULL;
    }
    return node;
}

// 前序遍历函数
void preOrder(TreeNode* root) {
    if (root == NULL) return;
    
    // 先访问根节点
    printf("%d\n", root->key);
    
    // 递归访问子节点
    for (int i = 0; i < 3; i++) {
        preOrder(root->children[i]);
    }
}

int main() {
    int rootKey, c1, c2, c3;
    // 读取4个节点的键值
    scanf("%d %d %d %d", &rootKey, &c1, &c2, &c3);
    
    // 创建树结构
    TreeNode* root = createNode(rootKey);
    root->children[0] = createNode(c1);  // 根的第一个孩子
    root->children[1] = createNode(c2);  // 根的第二个孩子
    root->children[2] = createNode(c3);  // 根的第三个孩子
    
    // 执行前序遍历
    preOrder(root);
    
    return 0;
}