给一个二叉树,返回中序遍历的节点的值。
Note:递归很简单,那么迭代呢?
方法一:递归
1.创建一个vector存结果
2.递归结束点:root == NULL
3.先左,再中,后右
Status:
Accepted
Runtime: 0 ms
class Solution {
public:
void fillres(TreeNode* root, vector<int> &result){
if(root == NULL) return;
fillres(root->left, result);
result.push_back(root->val);
fillres(root->right, result);
}
vector<int> inorderTraversal(TreeNode* root) {
vector<int> result;
fillres(root, result);
return result;
}
};
方法二:迭代
Status:
Accepted
Runtime: 0 ms
中序:左 -> 中(根) -> 右
栈:压中 -> 压左 -> 出左 -> 出中 -> 压右 -> 出右
压中压左:压入本节点的中,也是压入父节点的左 -> 再压入本节点的左,也是压入左子节点的中
出左出中:弹出本节点的左,也是弹出左子节点的中 -> 再弹出本节点的中,也是弹出父节点的左
压右:压入本节点的右,也是压入右子节点的中
出右:弹出本节点的右,也是弹出右子节点的中
1.创建一个vector保存结果
2.创建一个stack模拟递归压栈
3.while循环4、5、6、7
4.先把根节点root压入栈,再把所有root->left压入栈(想要先出左再出中,就要先压中再压左)
5.迭代结束点:stack.empty()
6.stack.top取一个点ptop,ptop既是父节点的left(先出左),也是它自己的root(先出左,现在出中),ptop存入vector
7.再访问ptop的right(先出左再出中,现在出右),continue
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> result;
// if(root == NULL) return result;
stack<TreeNode*> stk;
TreeNode* pNode = root;
while(1){
while(pNode){
stk.push(pNode);
pNode = pNode->left;
}
if(stk.empty()) break;
pNode = stk.top();
result.push_back(pNode->val);
stk.pop();
pNode = pNode->right;
}
return result;
}
};