zoukankan      html  css  js  c++  java
  • 根据所给树的头结点创建分层数组

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

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

    那么就又衍生了一个问题,就是保存在同一个索引的数组的前提是存在这么一个数组,意思就是,如果 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;
    }
  • 相关阅读:
    文件路径选择中的三态逻辑
    .net版本号
    使用MSBuild编译vs多个解决方案
    CEF截图
    使用SharpZIpLib写的压缩解压操作类
    软件试用期设置
    list转datatable
    excel 导入
    网站登录简单验证码
    UEditor编辑器
  • 原文地址:https://www.cnblogs.com/wuOverflow/p/4676834.html
Copyright © 2011-2022 走看看