题目描述
给定一个二叉树,返回它的 后序 遍历。
示例:
输入: [1,null,2,3] 1 2 / 3 输出: [3,2,1]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
解题思路
后序遍历的顺序是左孩子->右孩子->父节点,对于每个遍历到的节点,执行如下操作:
- 首先对该节点不断沿左孩子方向向下遍历并入栈,直到左孩子为空
- 取出栈顶节点,此时该节点为树的最左下节点,若其右孩子不为空,则回到步骤1;若为空说明其为叶子节点,将其值输出到结果中,并记录pre为当前节点
- 在步骤2判断右孩子是否为空时,同时判断当前节点右孩子是否已被输出,如果pre是其右孩子,说明当前节点的所有子节点已访问过,所以不必再向下遍历,直接输出即可
代码
1 /** 2 * Definition for a binary tree node. 3 * struct TreeNode { 4 * int val; 5 * TreeNode *left; 6 * TreeNode *right; 7 * TreeNode(int x) : val(x), left(NULL), right(NULL) {} 8 * }; 9 */ 10 class Solution { 11 public: 12 vector<int> postorderTraversal(TreeNode* root) { 13 vector<int> res; 14 TreeNode* node = root, *pre = NULL; 15 stack<TreeNode*> s; 16 while(node || s.size()){ 17 while(node){ 18 s.push(node); 19 node = node->left; 20 } 21 node = s.top(); 22 if(node->right && node->right != pre) 23 node = node->right; 24 else{ 25 res.push_back(node->val); 26 s.pop(); 27 pre = node; 28 node = NULL; 29 } 30 } 31 return res; 32 } 33 };
原文地址:https://www.cnblogs.com/wmx24/p/9494486.html
时间: 2024-10-09 11:02:03