1020. Tree Traversals (25)

1020. Tree Traversals (25)

题目

给定一棵二叉树的后续遍历和中序遍历的结果，输出层序遍历的结果。

思路：
后序结果的最后一个元素是根节点。在中序遍历结果中找到根节点，那么根节点左边的所有元素都是左子树结点（m个），右边的所有元素都是右子树结点（n个）。那么后续遍历的前m个元素都是左子树的结点，m+1~m+1+n个元素都是右子树的结点。按照这样的规律递归遍历中序和后序的数组。
定义strcut（层数，id），定义一个用于排序的vector。
每递归一次，当前root节点所在层数+1。每次访问一个结点，记录它的层数，放入vector中。在递归的时候按照先序访问结点，保证左边节点先访问，右边结点后访问。这样得到的vector能保证同一层结点的左右顺序是正确的（虽然vector中的元素不是按照层序排列的）。最后按照层数对vector中的元素稳定排序。