预设
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
此处仅列出迭代写法,递归写法暂不赘述。
首先拿前序遍历为例:
我们需要依照【根-左-右】的顺序进行遍历元素。
由此,需要对整棵树进行拆解,拆分成一个个结构相同的单元以便于我们统一处理。我们的单元就是一个仅仅包含三个节点的【根-左-右】。
但并不是所有存有值的节点都符合,如叶节点。为了相对便于统一处理,我们做这样一个假设:
给所有的叶子节点都补上null孩子,将这棵树填充“完满”。
从cur = root开始,进行遍历,以每个cur为根,逐个考察单元结构。
此时,我们将会对所有的cur进行直接考察,而不会考察cur->left或者是cur-right,以免混乱。
————————————————————————————————————————
由于按照【根-左-右】的顺序,故向左孩子移动。由于右孩子还没考察,故根节点不能随意放弃,应存入一个栈结构,以便之后使用。
我们搜寻过的cur元素值,都可以按顺序直接放入结果容器中。
直到当前节点为null,则说明【根-左】已完成,理应考察 上一个节点的右孩子。
// 即说明了我们需要保存曾经考察过的节点。
取出栈顶元素,将其右孩子作为新的cur考察。同时将该栈顶元素弹出。
// 该单元结构已经完成,无需继续存储。
只要栈内非空,或者当前cur不为null,即继续进行循环。
// 栈内为空,也可能需要循环(比如初始化时,栈就为空,cur为root;最后只剩一个右孩子时,栈为空,cur不为null)
// cur为null,自然更需要循环。只要叶子节点(或者是只有一侧孩子的根节点)为cur,那么马上就可能会碰到cur为null。而此时栈大概率未必为空。
前序遍历
class Solution {
public:
vector<int> T;
stack<TreeNode*> S;
vector<int> preorderTraversal(TreeNode* root) {
// 前序遍历
TreeNode* cur = root;
while (!S