Given a binary tree, return the inorder traversal of its nodes' values.
For example:
Given binary tree {1,#,2,3},
1
\
2
/
3
return [1,3,2].
这个题目的递归算法很简单:
class Solution {
public:
vector<int> inorderTraversal(TreeNode *root) {
vector<int> result;
inorderTraversal(root, result);
return result;
}
private:
void inorderTraversal(TreeNode *root, vector<int> &result) {
if (!root) return;
if (root->left ) inorderTraversal(root->left, result);
result.push_back(root->val);
if (root->right) inorderTraversal(root->right, result);
}
};
题目中要求不用递归的方法,当我思考如何用迭代的方法解决这个问题的时候,我发现这不是一个简单的问题,借助于从不让我失望的Discuss,学习到了如下非常巧妙的解法:
- 从根节点开始,先一直往树的左根部走,将这个过程中所经过的节点依次入栈;
- 当走到左根部的时候,从栈顶pop一个节点出来,输出该节点,接下来扫描该节点的右子树;
- 按照第1步的方法,先走到该右子树的左根部,输出左根部的节点,并扫描其右子树;
- 循环迭代该过程。
代码实现非常简洁:
vector<int> inorderTraversal(TreeNode *root) {
vector<int> result;
stack<TreeNode *> walker;
TreeNode *pCurNode = root;
while (!walker.empty() || pCurNode != nullptr) {
if (pCurNode != nullptr) {
walker.push(pCurNode);
pCurNode = pCurNode->left;
}
else {
pCurNode = walker.top();
walker.pop();
result.push_back(pCurNode->val);
pCurNode = pCurNode->right;
}
}
return result;
}