给定一个二叉树,返回它的 前序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [1,2,3]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-preorder-traversal
方法:使用栈模拟系统栈,写出非递归程序。
按照前序遍历思想,我们要对一个节点执行三个操作,首先打印该节点,然后遍历左右子树。那么入栈则是右子树先入栈,左子树入栈,打印节点。
这里我们新定义了一个结构Command,一个是判断当前状态,一个存放节点。
(1)首先判断当前二叉树是否为空,如果为空,直接返回定义的数组res。
(2)定义一个类型为Command的栈;
(3)初始化栈为 访问第一个节点;
(4)进入循环,直到栈为空,结束循环
(5)首先得到栈顶元素,然后弹出栈。
(6)判断是否是打印,如果是打印该节点,具体操作是将节点的值弹出放到数组中;
(7)否则就是访问节点的操作,为安全起见,使用assert函数。然后判断右、左孩子是否存在,存在放入栈中。最后放入打印该节点的操作做。
(8)循环结束后返回定义好的数组即可。
对于中序遍历和后序遍历只是在(7)步的顺序做了调整。
代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
struct Command{
string s;
TreeNode *node;
Command(string s, TreeNode *node) : s(s), node(node) {}
};
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> res;
if(root == NULL) return res;
stack<Command> stack;
stack.push(Command("go", root)); // 初始化放入第一个节点
while(!stack.empty()){
Command command = stack.top();
stack.pop();
if(command.s == "print")
res.push_back(command.node->val);
else{
assert(command.s == "go");
if(command.node->right)
stack.push(Command("go", command.node->right));
if(command.node->left)
stack.push(Command("go", command.node->left));
stack.push(Command("print", command.node));
}
}
return res;
}
};