• 根据所给树的头结点创建分层数组


     这个问题对我而言,难点就是当我用递归函数遍历各个结点时,不同子树的同一层结点怎样让他们保存在同一个容器里。

    保存在同一个容器里说明他们有共同的属性,没错,就是层数。而层数又相当于就是返回值的索引,那就简单了,就让他们存在索引相同的数组里。

    那么就又衍生了一个问题,就是保存在同一个索引的数组的前提是存在这么一个数组,意思就是,如果 vector 的 size 是 2, 可是这是第五层的结点,怎么存进 [4] 里呢?

    提前就把一些空 vecotor<int> 装进去怎么样?可是装多少合适呢?如果多了,怎样压缩呢?如果少了,又采用怎样的扩张策略呢?

    所以我就用了一个 “惰性” 的方法,就是如非必要,勿增数组,只有当是在是不够了,才会懒洋洋地 push 一个空数组用来盛放该层的结点。其不好的地方就是每次都要先调用 size() 判容量。其实容量也可以就使用一个静态变量来存储,每次只需要和这个变量比较,节省了函数调用的开销,但想起一个编程风格的书说尽可能少用全局变量,以及向函数传入尽可能少的参数,就此作罢。

    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> levelOrderTraversal;
        
        function<void(TreeNode*, size_t)> traversal;
        traversal = [&levelOrderTraversal, &traversal](TreeNode* node, size_t levelNo){
            if (node == nullptr){
                return;
            }
            
            if(levelNo >= levelOrderTraversal.size()){
                levelOrderTraversal.push_back(vector<int>());
            }
            
            levelOrderTraversal[levelNo].push_back(node->val);
            ++levelNo;
            traversal(node->left,  levelNo);
            traversal(node->right, levelNo);
        };
        traversal(root, 0);
        return levelOrderTraversal;
    }
  • 相关阅读:
    【读书笔记】构建之法(CH7~CH8)
    【课后作业】软件创新
    【个人开发】词频统计
    【读书笔记】没有银弹
    【个人开发】词频统计-代码规范
    【个人开发】词频统计-文档设计
    GitBook 使用
    Android NDK 入门与实践
    Python 爬虫实战(一):使用 requests 和 BeautifulSoup
    手把手教你做个人 app
  • 原文地址:https://www.cnblogs.com/wuOverflow/p/4676834.html
Copyright © 2020-2023  润新知