这个问题对我而言,难点就是当我用递归函数遍历各个结点时,不同子树的同一层结点怎样让他们保存在同一个容器里。
保存在同一个容器里说明他们有共同的属性,没错,就是层数。而层数又相当于就是返回值的索引,那就简单了,就让他们存在索引相同的数组里。
那么就又衍生了一个问题,就是保存在同一个索引的数组的前提是存在这么一个数组,意思就是,如果 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;
}