二叉树遍历

2021-04-10  本文已影响0人  yanlong107

前言

在学习或工作中,总会遇到和算法相关的问题。
而二叉树在算法中是绕不过的一个场景。 这里介绍下二叉树的相关遍历方法。

二叉树遍历

class Solution {
public:
  void addTreeNode(vector<int> &treeVector, TreeNode* root) {
        if (root == NULL) {
            return ;
        }
        // 前序遍历,优先查找根节点
        treeVector.push_back(root->val);
        addTreeNode(treeVector, root->left);
        addTreeNode(treeVector, root->right);
    } 
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> treeVector;
        addTreeNode(treeVector, root);
        return treeVector;
    }    
};
class Solution {
public:

    void helper(vector<vector<int>> &res, TreeNode* root, int leval) {
            if (root == NULL) {
                return;
            }
            if (res.size() <= leval) {
                vector<int> tmp;
                res.push_back(tmp);
            }
            res[leval].push_back(root->val);
            helper(res, root->left, leval + 1);
            helper(res, root->right, leval + 1);
    }

    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> res;
        helper(res, root, 0);
        return res;
        
    }
};
上一篇 下一篇

猜你喜欢

热点阅读