代码随想录二刷 ｜二叉树 ｜144.二叉树的前序遍历

题目描述

144.二叉树的前序遍历

给你二叉树的根节点 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]

提示：

• 树中节点数目在范围 [0, 100] 内
• -100 <= Node.val <= 100

解题思路

递归三部曲：

1. 确定递归函数的参数和返回值
   因为要打印出前序遍历节点的数值，所以参数里需要传入vector来放节点的数值，除了这一点就不需要再处理什么数据了也不需要有返回值，所以递归函数返回类型就是void，代码如下：

   void traversal(TreeNode* cur, vector<int>& vec)
   

2. 确定终止条件
   当前遍历节点空了，就直接return

   if (cur == NULL) return;
   

3. 确定单层递归的逻辑
   前序遍历是中左右的循序，所以在单层递归的逻辑，是要先取中节点的数值，代码如下：

   vec.push_back(cur->val);
   if (cur->left) vec.push(cur->left, vec);
   if (cur->right) vec.push(cur->right, vec);
   

代码实现

递归法

class Solution {
public:
	void traversal(TreeNode* cur, vector<int>& vec) {
		if (cur == NULL) return;
		vec.push_back(cur->val);
		if (cur->left) traversal(cur->left, vec);
		if (cur->right) traversal(cur->right, vac);
	}

    vector<int> preorderTraversal(TreeNode* root) {
		vector<int> result;
		traversal(root, result);
		return result;
    }
};

迭代法

// 前序遍历：中左右
// 入栈顺序：右左中
class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
		vector<int> result;
		stack<TreeNode*> st;
		if (root != NULL) st.push(root);
		while (!st.empty()) {
			TreeNode* node = st.top();
			if (node != NULL) {
				st.pop();
				if (node->right) st.push(node->right);
				if (node->left) st.push(node->left);
				st.push(node);
				st.push(NULL);
			} else {
				st.pop();
				node = st.top();
				st.pop();
				result.push_back(node->val);
			}
		}
		return result;
    }
};