原题链接:(迭代法) 94. 二叉树的中序遍历
思路:
使用迭代法跟前序遍历不一样,并不能通过交换顺序来达成中序遍历
1.迭代法的中序遍历为左中右,只要用来存储遍历结点顺序的栈为空且叶子结点为空,就代表遍历结束,退出循环
2.先使用指针cur存储root,然后只要cur不为NULL 也就是不为叶子结点,那就将该结点push进栈中
3.如果cur为叶子结点,此时左子树遍历完成,开始处理中的逻辑
4.获取栈顶结点元素,将栈顶结点元素弹出,将获取到的结点的值push_back进result容器中,然后再将cur的右子树push进栈中进行新一轮中序遍历
5.遍历完成时,将result返回即可
全代码:
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
stack<TreeNode* >stackA;
vector<int> result;
TreeNode* cur = root;
while(cur != NULL || !stackA.empty())
{//如果栈为空且叶子结点为空,就代表遍历结束,退出循环
if(cur != NULL)
{//结点不为空,将结点push进栈内
stackA.push(cur);
//结点等于结点的左子树
cur = cur ->left;
}else
{//结点为空,获取栈顶元素
cur = stackA.top();
//将栈顶元素弹出
stackA.pop();
//将获取到的元素的值push进容器中
result.push_back(cur ->val);
//结点等于结点的右子树
cur = cur ->right;
}
}
//返回容器
return result;
}
};