给定一个二叉树,返回它的中序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [1,3,2]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
思路1:递归法(标准模板)
class Solution {
public:
vector<int> ans;
vector<int> inorderTraversal(TreeNode* root) {
if (root != NULL) {
inorderTraversal(root->left);
ans.push_back(root->val);
inorderTraversal(root->right);
}
return ans;
}
};
思路2:迭代法(二叉树的非递归遍历)
1.定义一个栈st,根节点入栈。
2.如果栈顶元素存在左孩子,则一直向左,左孩子入栈,直到不存在左孩子的节点。
3.如果当前节点不存在左孩子,栈顶元素出栈压入答案数组,右孩子入栈,继续回到2判断,直至栈空。
AC代码:(C++):
vector<int> inorderTraversal(TreeNode* root) {
vector<int> ans;
stack<TreeNode*> st;
TreeNode* temp = root;
while (temp || !st.empty()) {
while (temp) { //如果存在左孩子,则一直向下入栈
st.push(temp);
temp = temp->left;
}
temp = st.top(); //取栈顶并出栈
st.pop();
ans.push_back(temp->val);
temp = temp->right; //如果存在右孩子,右孩子入栈,继续判断
}
return ans;
}