Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).
For example:
Given binary tree [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
return its level order traversal as:
[
[3],
[9,20],
[15,7]
]
这道题给定一棵二叉树,要求按照层数输出每一层的int型元素。
定义一个vector<vector<int>>型的静态变量ret作为返回值,定义函数levelorder执行每个root节点的具体操作。
初始level为0,判定ret的size()是否等于当前输入的level值,若相等,建立一个空的vector储存这一层的int值,将该节点的val值存入ret[level]中,接下来对root节点的左右节点分别执行levelorder函数操作,此时输入的level值加1。
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> ret;
levelorder(root, 0 , ret);
return ret;
}
void levelorder(TreeNode* root, int level, vector<vector<int>>& ret){
if(!root) return;
if(ret.size()==level) ret.push_back({});
ret[level].push_back(root->val);
levelorder(root->left, level+1, ret);
levelorder(root->right, level+1, ret);
}
};