144. 二叉树的前序遍历(二叉树)(Leetcode刷题笔记)
欢迎大家访问我的GitHub博客
https://lunan0320.cn
题目
给你二叉树的根节点 root ,返回它节点值的 前序 遍历。
示例 1:
输入:root = [1,null,2,3]
输出:[1,2,3]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [1]
输出:[1]
示例 4:
输入:root = [1,2]
输出:[1,2]
示例 5:
输入:root = [1,null,2]
输出:[1,2]
解题代码 C++ (递归)
解题思路:递归的思路遍历二叉树是比较适用的,在前、中、后序遍历的过程中都可以用到。
先判断终止条件,再就先push当前值,再递归左子树,再递归右子树即可。
对于其他顺序的遍历只要把递归的顺序换换就可以。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
//前序遍历
void traversal(TreeNode* root, vector<int>& res){
//终止条件
if(!root) return;
//中 左 右
res.push_back(root->val);
traversal(root->left, res);
traversal(root->right, res);
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int>res;
//遍历过程
traversal(root,res);
return res;
}
};
解题代码 C++(迭代法)
解题思路: 非递归的方式实现二叉树的前序遍历其实是用栈的结构。 递归就是一种栈,用栈去迭代也是完全可以的。
思路:先让根结点入栈,然后不停循环,直到栈为空,循环取出栈顶元素,将栈顶元素入vector,然后依次右、左子树入栈。
注意:这里是右、左子树,因为栈的弹出的时候,顺序才会变成左、右子树,这个细节需要注意。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> res;
//先判断首元素
if(!root) return res;
//根结点不为空,就入栈
st.push(root);
//不断的循环,直到栈为空
while(!st.empty()){
//取出栈顶元素
TreeNode* tmp = st.top();
st.pop();
//栈顶元素入vector
res.push_back(tmp->val);
//对于不为空的左右结点也入栈(注意顺序:先右后左才符合栈弹出的情况)
if(tmp->right) st.push(tmp->right);
if(tmp->left) st.push(tmp->left);
}
return res;
}
};
算法效率
执行用时: 0 ms , 在所有 C++ 提交中击败了 100.00% 的用户
内存消耗了 8.2 MB , 在所有 C++ 提交中击败了58.29% 的用户