二叉树的中序遍历
题目描述
中序遍历
思路
递归
题目还说递归太简单啦,你再用个迭代吧。
代码
- 递归
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> ans;
if (root!=nullptr)
Traversal(ans,root);
return ans;
}
void Traversal(vector<int> & ans,TreeNode* root){
if(root->left!=nullptr)
Traversal(ans,root->left);
ans.emplace_back(root->val);
if(root->right!=nullptr)
Traversal(ans,root->right);
}
};
- 迭代
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector <int> ans;
stack <TreeNode*> s;
TreeNode* t=root;
while(t!=nullptr||!s.empty()){
if(t!=nullptr){
if(t->left==nullptr){
ans.emplace_back(t->val);
t=t->right;
}
else{
s.emplace(t);//划重点:效率绝绝子,减少一次构造函数
t=t->left;
}
}
if(t==nullptr&&!s.empty()){
t=&s.top();
s.pop();
ans.emplace_back(t->val);
t=t->right;
}
}
return ans;
}
};
- 方法三:Morris 中序遍历
其实整个过程我们就多做一步:假设当前遍历到的节点为 x,将 x 的左子树中最右边的节点的右孩子指向 x,这样在左子树遍历完成后我们通过这个指向走回了 x,且能通过这个指向知晓我们已经遍历完成了左子树,而不用再通过栈来维护,省去了栈的空间复杂度。
感觉有点像线索树。
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> res;
TreeNode *predecessor = nullptr;
while (root != nullptr) {
if (root->left != nullptr) {
// predecessor 节点就是当前 root 节点向左走一步,然后一直向右走至无法走为止
predecessor = root->left;
while (predecessor->right != nullptr && predecessor->right != root) {
predecessor = predecessor->right;
}
// 让 predecessor 的右指针指向 root,继续遍历左子树
if (predecessor->right == nullptr) {
predecessor->right = root;
root = root->left;
}
// 说明左子树已经访问完了,我们需要断开链接
else {
res.push_back(root->val);
predecessor->right = nullptr;
root = root->right;
}
}
// 如果没有左孩子,则直接访问右孩子
else {
res.push_back(root->val);
root = root->right;
}
}
return res;
}
};