📟作者主页:慢热的陕西人
🌴专栏链接:力扣刷题日记
📣欢迎各位大佬👍点赞🔥关注🚓收藏,🍉留言
牛客热题:二叉树的后序遍历
题目链接
二叉树的后序遍历_牛客题霸_牛客网 (nowcoder.com)
方法一:递归
思路
``_postorder:函数是一个辅助函数,用于递归地遍历二叉树并将遍历结果存入
res` 中。
- 函数首先判断当前节点是否为空,如果为空则直接返回。
- 然后递归地对左子树进行后序遍历。
- 接着递归地对右子树进行后序遍历。
- 最后将当前节点的值加入
res
中。
代码
vector<int> res;
void _postorder(TreeNode* root)
{
if(root == nullptr) return ;
if(root->left != nullptr) _postorder(root->left);
if(root->right != nullptr) _postorder(root->right);
res.push_back(root->val);
}
vector<int> postorderTraversal(TreeNode* root)
{
_postorder(root);
return res;
}
复杂度
时间复杂度:O(N) ,遍历了一遍二叉树
空间复杂度:O(N), 创建了一个和二叉树节点数相等的数组,用来返回对应的答案
方法二:非递归
思路
代码
vector<int> postorderTraversal(TreeNode* root)
{
vector<int> result;
if (root == nullptr) return result;
stack<TreeNode*> nodeStack;
TreeNode* prevNode = nullptr; // 用于记录上一个访问过的节点
while (root != nullptr || !nodeStack.empty()) {
while (root != nullptr) {
nodeStack.push(root);
root = root->left;
}
root = nodeStack.top();
if (root->right == nullptr || root->right == prevNode) {
result.push_back(root->val);
nodeStack.pop();
prevNode = root;
root = nullptr; // 将 root 置为空,防止重复访问
} else {
root = root->right;
}
}
return result;
}
复杂度
时间复杂度:O(N),遍历了一遍二叉树
空间复杂度:O(N), 使用了额外的栈空间