Binary Tree Preorder Traversal 二叉树的前序遍历

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

 示例:

输入: [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;
    }




猜你喜欢

转载自blog.csdn.net/qq_26410101/article/details/80556979