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

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

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

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