解题思路:
中序遍历的顺序是:左根右。所以需要使用一个栈来存储根节点(每一次push操作时,都相当于把该节点当成根节点了,然后再向其左节点探索)。
程序中的大循环是:!s.empty() || root != NULL。栈非空,说明仍有根节点未被push_back进结果容器中,当栈为空时,说明最后一个根节点(逐个push上的最后一个,是指逻辑顺序上的最后一个)及其左子树的节点均被访问和输出,但其右子树上的节点可能还没被输出。所以要判断当前节点是否为空,即:root != NULL。
循环体内的程序逻辑:当左节点不为空时,一直向左探索,遇到NULL,则将栈顶元素输出,并将当前节点指向其右节点,访问右子树。
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> res;
if(root == NULL) return {};
stack<TreeNode*> s;
while(!s.empty() || root != NULL){ //当栈不为空,或者当前元素不为NULL时
while(root != NULL){ //当左节点不为NULL时,一直向左探索
s.push(root);
root = root->left;
}
//此时root节点为NULL了
root = s.top();
s.pop();
res.push_back(root->val);
root = root->right;
}
return res;
}
};