【PAT甲级】1174 Left-View of Binary Tree

• 问题思考： 

一眼层序遍历，取每层的第一个元素。

层序遍历需要有一颗二叉树，又给了中序遍历和前序遍历，需要构造一颗二叉树，套模板即可。但需要额外注意，自己写的时候，pre的边界容易写错。

层序遍历用一个队列实现，那么随着队列元素不断弹出，如何分辨此时temp到了下一层呢？

给节点加一个level属性，入队的元素肯定在temp节点的下一层。每次入队的时候计算节点的level

用一个cur变量记录当前已经取到第几层了，cur从0开始。

• 代码实现：

#include
#include
#include
using namespace std;
struct node {// 不熟悉静态写法，乖乖定义数据结构就好了
    int data, level;
    struct node *lchild, *rchild;
};
vector res;
int n, in[25], pre[25];
node *build(int in_arr[], int in_l, int in_r, int pre_arr[], int pre_l, int pre_r) {
    if (in_l > in_r) return nullptr;
    int pos = in_l;
    while (in_arr[pos] != pre_arr[pre_l] && pos <= in_r) pos++;
    node *root = new node;
    root->data = in_arr[pos];
    root->lchild = build(in_arr, in_l, pos - 1, pre_arr, pre_l + 1, pre_l + pos - in_l);
    // root->rchild = build(in_arr, pos + 1, in_r, pre_arr, pos + 1, pre_r);    // 错误示范，pre跟着pos走，肯定要走偏
    root->rchild = build(in_arr, pos + 1, in_r, pre_arr, pre_l + pos - in_l + 1, pre_r);     // 正确姿势，pre有它自己的基准
    return root;
}
int main() {
    srand(time(0));
    scanf("%d", &n);
    for (int i = 0; i < n; i++) {
        scanf("%d", in + i);
    }
    for (int i = 0; i < n; i++) {
        scanf("%d", pre + i);
    }
    node *root = build(in, 0, n - 1, pre, 0, n - 1);
    queue que;
    root->level = 0;
    que.push(root);
    res.push_back(root->data);
    int cur = 0;
    while (que.size()) {
        node *temp = que.front();
        que.pop();
        if (temp->level > cur) {
            res.push_back(temp->data);
            cur++;
        }
        if (temp->lchild) {
            temp->lchild->level = temp->level + 1;
            que.push(temp->lchild);
        }
        if (temp->rchild) {
            temp->rchild->level = temp->level + 1;
            que.push(temp->rchild);
        }
    }
    for (int i = 0; i < res.size(); i++) {
        i && printf(" ");
        printf("%d", res[i]);
    }
    return 0;
}