目录
一,题目描述
英文描述
Given the root of a binary tree, return the preorder traversal of its nodes' values.
中文描述
给你二叉树的根节点
root
,返回它节点值的 前序 遍历。
示例与说明
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-preorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二,解题思路
之前做过一题,是关于后序遍历的@&再见萤火虫&【LeetCode_Stack_145. Binary Tree Postorder Traversal 二叉树的后序遍历(Java)【栈,迭代】】
主要是借助栈和pre指针。
栈负责记录节点的路径,pre记录上一个遍历的节点(判断右子树是否已经遍历过)
三,AC代码
C++
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<TreeNode*> stack;
vector<int> ans;
if (root == nullptr) return ans;
TreeNode* temp = root->left, * pre = nullptr;
stack.push_back(root);
ans.push_back(root->val);
while (!stack.empty()) {
while (temp != nullptr) {
stack.push_back(temp);
ans.push_back(temp->val);
pre = temp;
temp = temp->left;
}
while (!stack.empty() && (stack.back()->right == nullptr || stack.back()->right == pre)) {
pre = stack.back();
stack.pop_back();
}
if (!stack.empty()) temp = stack.back()->right;
}
return ans;
}
};
Java
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> ans = new ArrayList<>();
Deque<TreeNode> stack = new LinkedList<>();
if (root == null) return ans;
TreeNode pre = new TreeNode(), temp = root.left;
stack.push(root);
ans.add(root.val);
while (!stack.isEmpty()) {
while (temp != null) {
stack.push(temp);
ans.add(temp.val);
pre = temp;
temp = temp.left;
}
while (!stack.isEmpty() && (stack.peek().right == null || stack.peek().right.equals(pre))) {
pre = stack.pop();
}
if (!stack.isEmpty()) temp = stack.peek().right;
}
return ans;
}
}
四,解题过程
第一博
虽然之前做过类似的一题,这次调试细节还是花了很久。。。