力扣 94. 二叉树的中序遍历

中序遍历的节点顺序为,左->中->右。本题使用了迭代法和递归法两种方法。

迭代法是使用的堆栈来模仿递归遍历算法中通常使用的函数调用堆栈。迭代法的中序遍历是左中右,需要先到达树左面的最底部,再开始处理节点。

递归法的关键点有三点,分别是确定递归函数的参数和返回值,确定终止条件,确定单层递归的逻辑。

迭代法时间复杂度:O(n),空间复杂度:O(n)

迭代法思路:

  1. 创建一个堆栈以使用非递归的方式,创建一个指针cur,初始化为root,该指针的目的是用来遍历树,同时跟踪正在处理的当前节点。
  2. 创建一个循环,该循环一直持续到指针cur变为NULL且堆栈变空,即全部节点处理完为止。
    1. 当cur指针不是NULL时,当前节点会尽量向左子节点移动,将遇到的所有节点(沿着最左边的路径)压入堆栈。这对应于中序遍历中的“左”步骤。这样在第一次弹出节点前,指针cur就到了树左面的最底部,再开始处理节点。
    2. 当cur指针是NULL时,表示已经到达最左边的节点。此时,我们从堆栈中弹出一个节点,该节点代表当前要处理的节点(当前子树中最左边的节点)。将当前节点的值添加到result中。栈中下一次循环中弹出的节点就是2.1中一直向左移动时遇到的节点中的中节点,这表示中序遍历中的“中间”步骤。
    3. 指针cur被更新为指向当前节点的右子节点(如果存在)。这对应于中序遍历中的“右”步骤。
  3. 循环完成后,返回result。

迭代法代码:

/**
 * 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> result;
        stack<TreeNode*> st;
        TreeNode* cur = root;
        while (cur != NULL || !st.empty()) {
            if (cur != NULL) { // 指针来访问节点,访问到最底层
                st.push(cur); // 将访问的节点放进栈
                cur = cur->left;                // 左
            } else {
                cur = st.top(); // 从栈里弹出的数据,就是要处理的数据(放进result数组里的数据)
                st.pop();
                result.push_back(cur->val);     // 中
                cur = cur->right;               // 右,退回到中节点后下一步就是这步深入到右子节点
            }
        }
        return result;
    }
};

递归法时间复杂度:O(n),空间复杂度:O(n)

递归法思路:

  1. 在本题的中序遍历中,创建一个vector<int>类型的参数,在该vector中存放遍历节点的数值,最后返回。
  2. 确定递归函数的参数和返回值:在递归函数中,因为要存放中序遍历节点的数值,所以参数里需要传入vector的指针或者引用来放节点的数值,不需要返回值,所以递归函数返回类型为void。
  3. 确定终止条件:在递归函数中,如果当前遍历的这个节点是个空节点,则直接return。
  4. 确定单层递归的逻辑:中序遍历是左中右的循序,所以在递归函数的逻辑中,先取递归取左节点的数值,再取中节点的数值,再递归取右节点的数值。
  5. 返回存放结果vector<int>。

递归法代码:

/**
 * 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:
    void traversal(TreeNode *node, vector<int> &ret)
    {
        if (node == NULL)
            return;
        traversal(node->left, ret);
        ret.push_back(node->val);
        traversal(node->right, ret);
    }
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> ret;
        traversal(root, ret);
        return ret;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值