A66705. 同上题中的二叉树，阅读以下二叉树的深度优先搜索代码:1 #include <iostream>

同上题中的二叉树，阅读以下二叉树的深度优先搜索代码:

1 #include <iostream>
2 #include <stack>
3
4 using namespace std;
5
6 // 非递归深度优先搜索（DFS）
7 TreeNode* dfs(TreeNode* root, int a) {
8  if (root == nullptr) return nullptr;
9
10  stack<TreeNode*> stk;
11  stk.push(root);
12
13  while (!stk.empty()) {
14   TreeNode* node = stk.top();
15   stk.pop();
16   if (node->val == a)
17    return node;
18
19   cout << node->val << " "; // 访问当前节点
20
21   if (node->right) stk.push(node->right); // 先压入右子节点
22   if (node->left) stk.push(node->left); // 再压入左子节点
23  }
24  return nullptr;
25 }