参考自:二叉树前序、中序、后序遍历相互求法(凡程子)
今天我们关注的依旧是二叉树的基础,前序、中序、后序遍历之间的关系。
一、已知前序、中序遍历,求后序。
举例说明:
前序遍历: GDAFEMHZ
中序遍历: ADEFGHMZ
画树求法:
第一步,根据前序遍历的特点,我们知道根结点为G;
第二步,观察中序遍历ADEFGHMZ。其中root节点G左侧的ADEF必然是root的左子树,G右侧的HMZ必然是root的右子树;
第三步,观察左子树ADEF,左子树的根节点必然是大树的root的leftchild。在前序遍历中,大树的root的leftchild位于root之后,所以左子树的根节点为D;
第四步,同样的道理,root的右子树节点HMZ中的根节点也可以通过前序遍历求得。在前序遍历中,一定是先把root和root的所有左子树节点遍历完之后才会遍历右子树,并且遍历的右子树的第一个节点就是右子树的根节点。可知,右子树的根节点为M;
第五步,观察发现,上面的过程是递归的。先找到当前树的根节点,然后划分为左子树,右子树,然后进入左子树重复上面的过程,然后进入右子树重复上面的过程。最后就可以还原一棵树了。
由前序和中序遍历直接递归求解后序遍历的过程可以简洁表达如下:
1 、确定根,确定左子树,确定右子树。
2、 在左子树中递归。
3、 在右子树中递归。
4、 打印当前根。
那么,我们可以画出这个二叉树的形状:
然后可知其后序遍历为:AEFDHZMG
依据前序、中序构建树:https://leetcode.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/
依照原题目使用 vector<int> 的方法得到,结果内存超了。
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
if(preorder.size() != inorder.size()) return NULL;
if(0 == preorder.size()) return NULL;
int leftLen, rightLen;
vector<int>::iterator rootPos = find(inorder.begin(), inorder.end(), preorder[0]);
leftLen = rootPos - inorder.begin();
rightLen = inorder.size() - 1 - leftLen;
TreeNode *root = new TreeNode(preorder[0]);
vector<int> leftPreorder, leftInorder, rightPreorder, rightInorder;
leftPreorder.assign(preorder.begin() + 1, preorder.begin() + 1 + leftLen);
leftInorder.assign(inorder.begin(), inorder.begin() + leftLen);
rightPreorder.assign(preorder.begin() + 1 + leftLen, preorder.end());
rightInorder.assign(inorder.begin() + 1 + leftLen, inorder.end());
root -> left = buildTree(leftPreorder, leftInorder);
root -> right = buildTree(rightPreorder, rightInorder);
return root;
}
换做下标,代码通过。
class Solution {
public:
TreeNode* buildSubTree(vector<int>& preorder, vector<int>& inorder, int posPre, int posIn, int len) {
if(0 == len) return NULL;
int leftLen, rightLen;
vector<int>::iterator rootPos = find(inorder.begin(), inorder.end(), preorder[posPre]);
leftLen = rootPos - (inorder.begin() + posIn);
rightLen = len - 1 - leftLen;
TreeNode *root = new TreeNode(preorder[posPre]);
root -> left = buildSubTree(preorder, inorder, posPre + 1, posIn, leftLen);
root -> right = buildSubTree(preorder, inorder, posPre + 1 + leftLen, posIn + 1 + leftLen, rightLen);
return root;
}
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
if(preorder.size() != inorder.size()) return NULL;
if(0 == preorder.size()) return NULL;
return buildSubTree(preorder, inorder, 0, 0, preorder.size());
}
};
二、已知中序、后序遍历,求前序。
用后序替换了前序,思路与一相似。
依然是上面的题,这次我们只给出中序和后序遍历:
中序遍历: ADEFGHMZ
后序遍历: AEFDHZMG
画树求法:
第一步,根据后序遍历的特点,我们知道后序遍历最后一个结点即为根结点,即根结点为G;
第二步,观察中序遍历ADEFGHMZ。其中root节点G左侧的ADEF必然是root的左子树,G右侧的HMZ必然是root的右子树;
第三步,观察左子树ADEF,左子树的中的根节点必然是大树的root的leftchild。在前序遍历中,大树的root的leftchild位于root之后,所以左子树的根节点为D;
第四步,同样的道理,root的右子树节点HMZ中的根节点也可以通过前序遍历求得。在前后序遍历中,一定是先把root和root的所有左子树节点遍历完之后才会遍历右子树,并且遍历的左子树的第一个节点就是左子树的根节点。同理,遍历的右子树的第一个节点就是右子树的根节点;
第五步,观察发现,上面的过程是递归的。先找到当前树的根节点,然后划分为左子树,右子树,然后进入左子树重复上面的过程,然后进入右子树重复上面的过程。最后就可以还原一棵树了。
由后序和中序遍历直接递归求解前序遍历的过程可以简洁表达如下:
1 确定根,确定左子树,确定右子树。
2 在左子树中递归。
3 在右子树中递归。
4 打印当前根。
这样,我们就可以画出二叉树的形状,如上图所示,这里就不再赘述。
那么,前序遍历: GDAFEMHZ
依据中序、后序构建树:https://leetcode.com/problems/construct-binary-tree-from-inorder-and-postorder-traversal/
class Solution {
public:
TreeNode* buildSubTree(vector<int>& inorder, vector<int>& postorder, int posIn, int posPost, int len) {
if(0 == len) return NULL;
int leftLen, rightLen;
vector<int>::iterator rootPos = find(inorder.begin(), inorder.end(), postorder[posPost + len - 1]);
leftLen = rootPos - (inorder.begin() + posIn);
rightLen = len - 1 - leftLen;
TreeNode *root = new TreeNode(postorder[posPost + len - 1]);
root -> left = buildSubTree(inorder, postorder, posIn, posPost, leftLen);
root -> right = buildSubTree(inorder, postorder, posIn + 1 + leftLen, posPost + leftLen, rightLen);
return root;
}
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
if(postorder.size() != inorder.size()) return NULL;
if(0 == postorder.size()) return NULL;
return buildSubTree(inorder, postorder, 0, 0, postorder.size());
}
};
三、已知前序、后序遍历,求中序。
这个不唯一。想想为什么?