利用栈和一个visit变量,visit用来记录当前节点是否被拜访过
#include <iostream>
#include <vector>
#include <algorithm>
#include <iterator>
#include <stack>
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};
// 递归
void PostorderTraversalRecursion(TreeNode* root, std::vector<int>& res)
{
if (root) {
PostorderTraversalRecursion(root->left, res);
PostorderTraversalRecursion(root->right, res);
res.push_back(root->val);
}
}
// 非递归
void PostorderTraversal(TreeNode* root, std::vector<int>& res)
{
std::stack<TreeNode*> s;
TreeNode* visit;
while (root) {
s.push(root);
root = root->left;
}
while (!s.empty()) {
root = s.top();
s.pop();
if (!root->right || root->right == visit) {
res.push_back(root->val);
visit = root;
}
else {
s.push(root);
root = root->right;
while (root) {
s.push(root);
root = root->left;
}
}
}
}
int main()
{
TreeNode* n1 = new TreeNode(1);
TreeNode* n2 = new TreeNode(2);
TreeNode* n3 = new TreeNode(3);
TreeNode* n4 = new TreeNode(4);
TreeNode* n5 = new TreeNode(5);
TreeNode* n6 = new TreeNode(6);
TreeNode* n7 = new TreeNode(7);
n1->left = n2;
n1->right = n5;
n2->left = n3;
n2->right = n4;
n5->left = n6;
n5->right = n7;
std::vector<int> res;
PostorderTraversalRecursion(n1, res);
std::copy(res.begin(), res.end(), std::ostream_iterator<int>(std::cout, " "));
std::cout << std::endl;
res.clear();
PostorderTraversal(n1, res);
std::copy(res.begin(), res.end(), std::ostream_iterator<int>(std::cout, " "));
std::cout << std::endl;
}