LeetCode_Stack_144. Binary Tree Preorder Traversal 二叉树的前序遍历【栈,迭代】【简单】

 

目录

一,题目描述

英文描述

中文描述

示例与说明

二,解题思路

三,AC代码

C++

Java

四,解题过程

第一博


一,题目描述

英文描述

Given the root of a binary tree, return the preorder traversal of its nodes' values.

中文描述

给你二叉树的根节点 root ,返回它节点值的 前序 遍历。

示例与说明

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-preorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

二,解题思路

之前做过一题,是关于后序遍历的@&再见萤火虫&【LeetCode_Stack_145. Binary Tree Postorder Traversal 二叉树的后序遍历(Java)【栈,迭代】】

主要是借助栈和pre指针。

栈负责记录节点的路径,pre记录上一个遍历的节点(判断右子树是否已经遍历过)

三,AC代码

C++

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        vector<TreeNode*> stack;
        vector<int> ans;
        if (root == nullptr) return ans;
        TreeNode* temp = root->left, * pre = nullptr;
        stack.push_back(root);
        ans.push_back(root->val);
        while (!stack.empty()) {
            while (temp != nullptr) {
                stack.push_back(temp);
                ans.push_back(temp->val);
                pre = temp;
                temp = temp->left;
            }
            while (!stack.empty() && (stack.back()->right == nullptr || stack.back()->right == pre)) {
                pre = stack.back();
                stack.pop_back();
            }
            if (!stack.empty()) temp = stack.back()->right;
        }
        return ans;
    }
};

Java

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> ans = new ArrayList<>();
        Deque<TreeNode> stack = new LinkedList<>();
        if (root == null) return ans;
        TreeNode pre = new TreeNode(), temp = root.left;
        stack.push(root);
        ans.add(root.val);
        while (!stack.isEmpty()) {
            while (temp != null) {
                stack.push(temp);
                ans.add(temp.val);
                pre = temp;
                temp = temp.left;
            }
            while (!stack.isEmpty() && (stack.peek().right == null || stack.peek().right.equals(pre))) {
                pre = stack.pop();
            }
            if (!stack.isEmpty()) temp = stack.peek().right;
        }
        return ans;
    }
}

四,解题过程

第一博

虽然之前做过类似的一题,这次调试细节还是花了很久。。。

 

Guess you like

Origin blog.csdn.net/qq_41528502/article/details/120403150