给定一个二叉树,返回它的 前序 遍历。
示例:
输入: [1,null,2,3] 1 \ 2 / 3 输出: [1,2,3]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
思路:
这道题有递归和非递归,先讲递归算法。
递归解法:在函数的参数中引入vector的引用,按照前序遍历(根-左-右),访问到当前节点就累加到vector中。最后返回vector
递归代码:
void preorderTraversalCore(TreeNode* root, vector<int> &res) { if (!root) { return; } res.push_back(root->val); preorderTraversalCore(root->left,res); preorderTraversalCore(root->right,res); } vector<int> preorderTraversal(TreeNode* root) { vector<int> res; if (!root) { return res; } preorderTraversalCore(root, res); return res; }
方法二(迭代代码):
这也是题目要求的前序遍历的非递归方法,我们需要掌握。
迭代遍历前序序列需要引入一个栈(先进后出特性),每次访问当前节点时压入栈,如果当前节点有左节点就不断压入栈,如果没有左节点意味着已经到最左叶节点,这时取栈顶元素pop出,如果pop的元素有右节点就压入stack,然后不断循环即可。
代码如下:
vector<int> preorderTraversal(TreeNode* root) { vector<int> res; if (!root) { return res; } stack<TreeNode*> stack; TreeNode* tmp = root; while (tmp != nullptr || !stack.empty()) { while (tmp) { res.push_back(tmp->val); stack.push(tmp); tmp = tmp->left; } if (!stack.empty()) { tmp = stack.top(); stack.pop(); tmp = tmp->right; } } return res; }