给你二叉树的根节点 root ,返回它节点值的前序遍历。
实例一:
输入:root = [1,null,2,3]
输出:[1,2,3]
解法一:采用递归来实现前序遍历的逻辑
根据递归的三要素:
1、确定递归函数的参数和返回值:确定哪些参数是递归过程中需要进行处理的,就需要在递归函数里加上这个参数,并且还要明确每次递归的返回值是什么进行确定递归函数的返回类型。
2、确定终止条件:递归算法,经常会遇到栈溢出的错误,就是终止条件写的有问题,操作系统是用一个栈结构来保存每一层递归的信息,如果递归没有终止,操作系统的内存栈就会溢出。
3、确定单层递归的逻辑:确定每一层递归需要处理的信息,这里就会重复调用自己来实现递归的过程。
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> result;
traversal(root, result);
return result;
}
void traversal(TreeNode* root, vector<int>& res)
{
if(root == nullptr)
{
return;
}
res.push_back(root->val);
traversal(root->left, res);
traversal(root->right, res);
}
};
解法二:采用非递归来实现:
用递归解决的方法都能用非递归的方法实现,因为递归方法是利用函数栈来保存信息,如果用自己申请的数据结构来代替栈,也可以实现相同的功能。
用非递归的方式来实现二叉树的先序遍历,具体过程如下:
1、申请一个新的栈stack,然后将头结点head压入栈中。
2、从stack中弹出栈顶节点,即为tmpNode,然后打印tmpNode的值,再将节点tmpNode的右孩子节点(不为空)先压入栈中,最后将tmpNode的左孩子(不为空)压入stack中。
3、不断重复1和2,直到栈为空,全部过程结束。
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
stack<TreeNode*> sck;
vector<int> result;
if(root != nullptr)
{
sck.push(root);
while(!sck.empty())
{
TreeNode* tmpNode = sck.top();
sck.pop();
result.push_back(tmpNode->val);
if(tmpNode->right) sck.push(tmpNode->right);
if(tmpNode->left) sck.push(tmpNode->left);
}
}
return result;
}
};