牛客网真题PAT【甲级】1010

题目链接：牛客网真题
题目简述：输入n，表示树中总结点数。而后给出n对数字，表示第i号节点的左右孩子的位置。最后给出一列无序的数字。让求这棵树的层序遍历序列。

#include<bits/stdc++.h>
using namespace std;

int n, lay = 0;
vector<vector<int>> v(100, vector<int>(2, 0)), level(100);
vector<int> value, inorder, ans;

void traverse1(int root){//中序遍历
    if(root == -1) return;
    traverse1(v[root][0]);
    inorder.push_back(root);
    traverse1(v[root][1]);
}

void traverse2(int root, int layer){//层序遍历
    if(root == -1) return;
    lay = max(lay, layer);//获取最大的层数
    level[layer].push_back(root);
    traverse2(v[root][0], layer + 1);
    traverse2(v[root][1], layer + 1);
}

int main(){
    cin >> n;
    value.resize(n);
    ans.resize(n);
    for (int i = 0; i < n;i++)
        cin >> v[i][0] >> v[i][1];
    for (int i = 0; i < n;i++)
        cin >> value[i];
    sort(value.begin(), value.end());//对所给关键字排序，以便匹配对应中序序列
    traverse1(0);
    traverse2(0, 0);
    for (int i = 0; i < n;i++)
        ans[inorder[i]] = value[i];//获取对应编号位置的关键字，例如：ans[0]表示0号节点的关键字
    cout << ans[level[0][0]];
    for (int i = 1; i <= lay;i++)
        for (int j = 0; j < level[i].size();j++)
            cout << " " << ans[level[i][j]];
    return 0;
}

这个题目其实还是比较简单的，不过关键是要注意到一个关键点：层序遍历序列中，有节点编号序列和关键字编号序列，显然获取编号序列要简单，最后可以再过渡到关键字序列。

所以现在将问题转化为：如何找到每个节点的关键字。

很自然，二叉搜索树便隐含了中序遍历序列。那么我们只需获得中序遍历的节点编号构成的序列就好了。故先进行中序遍历，获得节点编号的中序序列，而后与中序序列的关键字进行匹配，最终得到了ans这个数组，它记录了每个节点的关键字。