浙大PTA 7-4 列出叶结点 (20point(s))

 

7-4 列出叶结点 (20point(s))

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

输入格式：

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

输出格式：

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

输入样例：

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

输出样例：

4 1 5

题目要求：按照层序遍历的形式输出叶子节点

思路：按照输入建树，用check数组保存第i个节点是否为叶子节点（是否有左孩子，右孩子），进行层序遍历，若为叶子节点则输出，否则不输出。

#include 
#include
#include
using namespace std;
struct TNode
{
    int Data;
    int Left;
    int Right;
} T[20];

int main()
{
    int n;
    int check[11] = {0};
    scanf("%d", &n);
    getchar();
    for(int i = 0; i < n; i++)
    {
        char cl, cr;
        scanf("%c %c", &cl, &cr);
        if(cl == '-')
            T[i].Left = -1;
        else
        {
            T[i].Left = cl - '0';
            check[T[i].Left] = 1;
        }
        if(cr == '-')
            T[i].Right = -1;
        else
        {
            T[i].Right = cr - '0';
            check[T[i].Right] = 1;
        }
        getchar();
    }
    int root;
    for(int i = 0; i < n; i++)
        if(check[i] == 0)
        {
            root = i;
            break;
        }
    queueq;
    q.push(root);
    bool flag = true;
    while(!q.empty())
    {
        int tmp = q.front();
        q.pop();
        if(T[tmp].Left != -1)
            q.push(T[tmp].Left);
        if(T[tmp].Right!= -1)
            q.push(T[tmp].Right);
        if(T[tmp].Left == -1&&T[tmp].Right == -1)
        {
            if(flag){
                flag = false;
            }
            else
                printf(" ");
            printf("%d", tmp);
        }
    }
    return 0;
}