递归和非递归实现前序遍历
//144. 二叉树的前序遍历
//dfs,前序遍历模板
class Solution {
public:
vector<int> ans;
void dfs(TreeNode* root) {
if (root == nullptr) return;
ans.emplace_back(root->val);
dfs(root->left);
dfs(root->right);
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int>ans;
dfs(root);
return ans;
}
};
// 进阶要求:尝试使用迭代解决
// 使用栈模拟实现dfs(使用队列模拟bfs)
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int>ans;
if (root == nullptr) return ans;
stack<TreeNode*>stk;
auto node = root;
while (stk.size() || node != nullptr) {
while (node != nullptr) {//一直循环找node节点的左孩子
ans.emplace_back(node->val);
stk.push(node);
node = node->left;
}
//找到右孩子,继续循环找左节点
node = stk.top(); stk.pop();
node = node->right;
}
return ans;
}
};