牛客 BM24. 二叉树的中序遍历
原题链接
简单
作者:
Mr-Black
,
2023-07-15 23:15:08
,
所有人可见
,
阅读 79
题目描述
二叉树的中序遍历
算法1
递归 $O(n)$
C++ 代码
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @return int整型vector
*/
void interorder(vector<int> &res, TreeNode* root){
if (root == NULL) return;
interorder(res, root -> left);
res.push_back(root->val);
interorder(res, root -> right);
}
vector<int> inorderTraversal(TreeNode* root) {
// write code here
vector<int> res;
interorder(res, root);
return res;
}
};
算法2
栈 $O(n)$
C++ 代码
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> res;
//辅助栈
stack<TreeNode*> s;
//当树节点不为空或栈中有节点时
while(root != NULL || !s.empty()){
//每次找到最左节点
while(root != NULL){
s.push(root);
root = root->left;
}
//访问该节点
TreeNode* node = s.top();
s.pop();
res.push_back(node->val);
//进入右节点
root = node->right;
}
return res;
}
};