递归遍历前中后序区别仅在于visit(root)位置:前序根左右、中序左根右、后序左右根;非递归中序用栈模拟,统一迭代法以nullptr标记访问时机。
前序、中序、后序的区别不在逻辑结构,而在 visit(root) 这句代码放在递归调用的哪个位置。只要记住:前序是“根左右”,中序是“左根右”,后序是“左右根”,对应到代码就是访问节点的时机不同。
假设二叉树节点定义为:
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode() : val(0), left(nullptr), right(nullptr) {}
TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};三种遍历的递归写法仅差一行顺序:
// 前序:根 → 左 → 右 void preorder(TreeNode* root, vector& res) { if (!root) return; res.push_back(root->val); // ✅ 访问放最前 preorder(root->left, res); preorder(root->right, res); }
// 中序:左 → 根 → 右 void inorder(TreeNode* root, vector
& res) { if (!root) return; inorder(root->left, res); // ✅ 先递归左 res.push_back(root->val); // ✅ 访问在中间 inorder(root->right, res); } // 后序:左 → 右 → 根 void postorder(TreeNode* root, vector
& res) { if (!root) return; postorder(root->left, res); postorder(root->right, res); res.push_back(root->val); // ✅ 访问放最后 }
面试和工程中更常考非递归写法,尤其是中序——它天然对应 BST 的升序输出。关键不是背模板,而是理解栈里存的是“待回溯的根节点”,每次弹出即表示“左子树已处理完,现在该访问自己了”。
cur != nullptr || !stk.empty(),不能只判栈空cur 设为 cur->right,进入右子树处理vectorinorderIterative(TreeNode* root) { vector res; stack stk; TreeNode* cur = root; while (cur || !stk.empty()) { while (cur) { stk.push(cur); cur = cur->left; // 一路压左 } cur = stk.top(); stk.pop(); res.push_back(cur->val); // 访问 cur = cur->right; // 切换到右子树 } return res; }
想一套代码通吃三种遍历?可以,但代价是空间多存一倍指针。思路是把“该访问根节点”这个动作也压栈,用 nullptr 作为标记。遇到 nullptr 就说明下一个节点就是该被访问的根。
以前序为例(中/后序只需调整入栈顺序):
root->right,再 nullptr,再 root->left(保证左先出)nullptr 时,下一个弹出的就是要访问的节点!stk.empty() 再取顶,否则运行时崩溃vectorpreorderUniform(TreeNode* root) { vector res; stack stk; if (root) stk.push(root); while (!stk.empty()) { TreeNode* node = stk.top(); stk.pop(); if (node != nullptr) { if (node->right) stk.push(node->right); if (node->left) stk.push(node->left); stk.push(node); // 根放最后 stk.push(nullptr); // 标记:下次该访问它 } else { node = stk.top(); stk.pop(); res.push_back(node->val); } } return res; }
题目问的是“前中后序”,本质是深度优先搜索(DFS)的三种访问顺序。而层序(BFS)用队列,逻辑完全不同:queue + 每次取 q.size() 个节点处理一层。
容易踩的坑:
TreeNode 默认构造函数可能不带初始化,left/right 野指针导致 UBSAN 报错vector 还是直接打印?接口要求决定是否需要传引用或返回新容器真正难的不是写出来,是在边界 case(空树、单节点、只有左/右子树)下逻辑依然自洽。动手前先手画三节点树跑一遍指针流转,比硬背模板可靠得多。