Java教程

【数据结构】二叉树专题

本文主要是介绍【数据结构】二叉树专题,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

LeetCode 94. 二叉树的中序遍历

递归写法

/**
 * 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> res;
    void dfs(TreeNode* root)
    {
        if(!root) return;
        dfs(root->left);
        res.push_back(root->val);
        dfs(root->right);
    }

    vector<int> inorderTraversal(TreeNode* root) {
        dfs(root);
        return res;    
    }
};

作者:Once.
链接:https://www.acwing.com/activity/content/code/content/3201627/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

非递归写法(利用栈,固定套路,要求背会)

/**
 * 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> res; // 定义答案数组
        stack<TreeNode*> stk; // 定义栈

        while(root || stk.size()) // 当当前结点非空或栈非空时
        {
            while(root) // 当当前结点不空时
            {
                stk.push(root); // 将当前结点入栈
                root = root->left; // 当前结点走到其左儿子的位置
            }
            root = stk.top(); // 将栈顶元素的值取出
            stk.pop(); // 将栈顶元素弹出
            res.push_back(root->val); // 遍历当前点
            root = root->right; // 遍历完之后走到当前结点的右儿子
        }
        return res; // 返回答案数组
    }
};

作者:Once.
链接:https://www.acwing.com/activity/content/code/content/3201627/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
这篇关于【数据结构】二叉树专题的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!