题目
给出一棵二叉树,返回其节点值的后序遍历。
思路
后序比较麻烦 需要另外一个变量来记录当前节点入栈的次数
设计pair<TreeNode*, int> p;
p.first 为二叉树节点
p.second 为当前节点入栈的次数
C++代码
vector<int> postorderTraversal(TreeNode *root) {
// write your code here
vector<int> vec;
stack<pair<TreeNode*, int>> s;
s.push(make_pair(root,0));
while(root && !s.empty())
{
TreeNode* p = s.top().first;
int times = s.top().second;
s.pop();
if(0 == times) //如果第一次入栈
{
s.push(make_pair(p,1)); //第二次入
if(p->right) s.push(make_pair(p->right,0));
if(p->left) s.push(make_pair(p->left,0));
}
else vec.push_back(p->val);
}
return vec;
}