题目
给定一个二叉树,返回它的中序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [1,3,2]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
思路
参考【数据结构】二叉树(前、中、后)序遍历的递归与非递归算法
整体思路:
都借助于栈,第一个入栈的节点均为根节点,不同的是出栈和入栈的时机。
1.对于前序遍历,首先输出根节点,所以每轮只压入根节点,然后先将右子节点入栈,再将左子节点入栈,然后继续出栈,依次迭代到栈为空
2.对于中序遍历,首先输出最左边的子节点,所以先将根节点的左子节点循环入栈,出栈的时候如果有右子节点,再将右子节点入栈,依次迭代到栈为空
3.与2类似,但父节点是最后一个输出来,所以根据父节点找右子树时需要先将父节点保存下来避免陷入死循环,依次迭代到栈为空
实现
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> res;
stack<TreeNode*> temp;
while (!temp.empty() || root != NULL) {
//先将左子树入栈
while (root != NULL) {
temp.push(root);
root = root->left;
}
root = temp.top();
//存储根节点
res.push_back(root->val);
temp.pop();
if (root->right == NULL) {
//如果右子树为空,向上查找父节点
root = NULL;
} else {
//如果存在右子树,将右子树入栈
root = root->right;
}
}
return res;
}
};