题目
给你一棵二叉树的根节点 root
,返回其节点值的 后序遍历 。
示例 1:
输入:root = [1,null,2,3] 输出:[3,2,1]
示例 2:
输入:root = [] 输出:[]
示例 3:
输入:root = [1] 输出:[1]
进阶:递归算法很简单,你可以通过迭代算法完成吗?
方法一:普通递归写法
/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode() : val(0), left(nullptr), right(nullptr) {}* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/
class Solution {
public:void istree(TreeNode* root,vector<int>& arr2){if(root==nullptr)return;istree(root->left,arr2);//先遍历左子树istree(root->right,arr2);//遍历右子树arr2.push_back(root->val);//保存当前节点的值return;}vector<int> postorderTraversal(TreeNode* root) {vector<int> arr1;istree(root,arr1);return arr1;}
};
方法二:通过栈实现非递归迭代
/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode() : val(0), left(nullptr), right(nullptr) {}* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/
class Solution {
public:vector<int> postorderTraversal(TreeNode* root) {vector<int> arr1;stack<TreeNode*> st;TreeNode* cur=root;TreeNode* prev;while(cur||!st.empty())//cur不为空或栈里还有节点就继续{while(cur)//找左路并push,直到找到最左节点{st.push(cur);cur=cur->left;}TreeNode* top=st.top();//保存最左节点//如果右为空或者prev和该节点右子节点相等说明右边已经访问过了,可以popif(top->right==nullptr||prev==top->right){st.pop();prev=top;//记录此次访问节点地址,为下一次做参考arr1.push_back(top->val);}else{cur=top->right;}}return arr1;}
};