二叉树的后序遍历的递归算法很简单,就是按照左、右、根的顺序进行递归遍历。这里贴上我写的算法:
class Solution {
public:
vector<int> mVector;
vector<int> postorderTraversal(TreeNode* root) {
if(root==NULL)
return mVector;
else{
postorderTraversal(root->left);
postorderTraversal(root->right);
mVector.push_back(root->val);
}
return mVector;
}
};
下面采用的是栈和后序遍历的非递归算法,解释一下原理,就是后序遍历的话,也要记录每一个根节点,然后因为要按照左、右、根的顺序进行遍历,所以说肯定要让root等于栈顶元素之后才开始,这之后现在root已经是左子树的最末尾端了,所以说要看看它有没有右子树,如果它有右子树的话,并且右子树没有被访问过,应该先把右子树挪动到栈里,进行访问,否则,就直接访问这个节点,并且将这个节点赋值为空,下一次直接让它等于栈顶元素就OK了,一个小技巧是因为要记录是否已经访问右节点,如果没有访问肯定会优先访问右节点,所以要设置一个指针记录右节点的访问情况,记录下上一次访问节点的情况。如果访问了右节点,则直接访问根节点就可以了。
下面贴上代码:
class Solution {
public:
vector<int> mVector;
stack<TreeNode*> mStack;
TreeNode *r=NULL;
vector<int> postorderTraversal(TreeNode* root) {
if(root==NULL)
return mVector;
while(root!=NULL||!mStack.empty()){
if(root!=NULL){
mStack.push(root);
root=root->left;
}
else{
root=mStack.top();
if(root->right!=NULL&&r!=root->right){
root=root->right;
}
else{
mVector.push_back(root->val);
r=root;
mStack.pop();
root=NULL;
}
}
}
return mVector;
}
};