【PTA数据结构 | C语言版】列出叶结点

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

题目

对于给定的二叉树，本题要求你按从上到下、从左到右的顺序输出其所有叶结点。

输入格式：
首先第一行给出一个正整数 n（≤10），为树中结点总数。树中的结点从 0 到 n−1 编号。随后 n 行，每行给出一个对应结点左右孩子的编号。如果某个孩子不存在，则在对应位置给出 “-”。编号间以 1 个空格分隔。

输出格式：
在一行中按规定顺序输出叶结点的编号。编号间以 1 个空格分隔，行首尾不得有多余空格。

输入样例：

8
1 -
- -
0 -
2 7
- -
- -
5 -
4 6

输出样例：
4 1 5

代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>#define MAX_N 10typedef struct TreeNode {int left;int right;
} TreeNode;int main() {int n;scanf("%d", &n);getchar(); // 消耗换行符TreeNode nodes[MAX_N];int isRoot[MAX_N];memset(isRoot, 1, sizeof(isRoot)); // 初始化所有节点为根候选// 读取输入并构建树for (int i = 0; i < n; i++) {char left[2], right[2];scanf("%s %s", left, right);// 处理左子节点if (left[0] == '-') {nodes[i].left = -1;} else {nodes[i].left = atoi(left);isRoot[nodes[i].left] = 0; // 该节点有父节点，不可能是根}// 处理右子节点if (right[0] == '-') {nodes[i].right = -1;} else {nodes[i].right = atoi(right);isRoot[nodes[i].right] = 0; // 该节点有父节点，不可能是根}}// 确定根节点int root = -1;for (int i = 0; i < n; i++) {if (isRoot[i]) {root = i;break;}}// 层序遍历队列int queue[MAX_N];int front = 0, rear = 0;queue[rear++] = root;int leaves[MAX_N];int leafCount = 0;// 层序遍历while (front < rear) {int current = queue[front++];// 如果是叶节点if (nodes[current].left == -1 && nodes[current].right == -1) {leaves[leafCount++] = current;}// 将子节点加入队列if (nodes[current].left != -1) {queue[rear++] = nodes[current].left;}if (nodes[current].right != -1) {queue[rear++] = nodes[current].right;}}// 输出结果for (int i = 0; i < leafCount; i++) {printf("%d", leaves[i]);if (i < leafCount - 1) {printf(" ");}}printf("\n");return 0;
}