二叉树的中序遍历
给定一个二叉树,返回它的 前序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [1,3,2]
题解
思路:
中序遍历和前序遍历的不同在于入栈顺序不同,想要结果集按照左子树、根节点、右子树的顺序存储,那么遍历二叉树结点时就需要以又孩子、根节点、左孩子的顺序入栈。最先让根节点入栈(此时是它第一次入栈),在栈非空的情况下,用临时结点对象保存栈顶结点,然后栈顶结点出栈,接下来就可以顺着临时结点对象找到栈顶结点的右孩子(右孩子入栈),然后根节点入栈(紧随其后是一个空结点入栈,用以标志结点的第二次入栈),最后是左孩子入栈。若临时结点对象为空,这时候就说明轮到该空结点的前一个结点存入结果集res中了。如此循环直至栈空,整个中序遍历结束。过程中包括根结点在内的所有结点都是2次入栈 :1次是为了顺藤摸瓜,1次是为了存入结果集res。
vector<int> inorderTraversal(TreeNode* root) {
vector<int> res;
stack<TreeNode*> call;
if(root!=nullptr) call.push(root);
while(!call.empty()){
TreeNode *t = call.top();
call.pop();
if(t!=nullptr){
if(t->right) call.push(t->right);
call.push(t); //在左节点之前重新插入该节点,以便在左节点之后处理(访问值)
call.push(nullptr); //nullptr跟随t插入,标识已经访问过,还没有被处理
if(t->left) call.push(t->left);
}else{
res.push_back(call.top()->val);
call.pop();
}
}
return res;
}