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

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

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

那么就又衍生了一个问题，就是保存在同一个索引的数组的前提是存在这么一个数组，意思就是，如果 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;
}

 

转载于:https://www.cnblogs.com/wuOverflow/p/4676834.html