LeetCode 二叉树的前序遍历(递归、递推)

给定一个二叉树,返回它的 前序 遍历。
示例:

输入: [1,null,2,3]  
   1
    \
     2
    /
   3 
输出: [1,2,3]

进阶: 递归算法很简单,你可以通过迭代算法完成吗?
思路分析:
前序遍历:先根节点,后左子树,再右子树

方法一:最为常见的莫过于递归法啦。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
	vector result;
	vector preorderTraversal(TreeNode* root) {
		myPreorderTraversal(root);
		return result;
	}
	//前序遍历root
	void myPreorderTraversal(TreeNode* root) {
		if (root == NULL) {
			return;
		}
		result.push_back(root->val);//先访问根节点
		myPreorderTraversal(root->left);//再访问左子树
		myPreorderTraversal(root->right);//后访问右子树
	}
};

方法二:利用栈的辅助,非递归实现。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
	vector result;
	vector preorderTraversal(TreeNode* root) {
		if (root == NULL) {
			return {};
		}
		stack myStack;
		while (root != NULL || !myStack.empty()) {
			if (root != NULL) {
				result.push_back(root->val);//先访问根节点
				if (root->right) {//保留右子树的现场
					myStack.push(root->right);
				}
				root = root->left;//转移现场至左子树
				continue;
			}
			root = myStack.top();//恢复现场
			myStack.pop();
		}
		return result;
	}
};

LeetCode 二叉树的前序遍历(递归、递推)_第1张图片

你可能感兴趣的:(LeetCode)