基础篇
树的层次遍历
- 算法描述:
特例处理: 当根节点为空,则返回空列表 [] ;
初始化: 打印结果列表 res = [] ,包含根节点的队列 queue = [root] ;
BFS 循环:
当队列 queue 为空时跳出;
新建一个临时列表 tmp ,用于存储当前层打印结果;
当前层打印循环: 循环次数为当前层节点数(即队列 queue .size()长度);
出队: 队首元素出队,记为 node;
打印: 将 node.val 添加至 tmp 尾部;
添加子节点: 若 node 的左(右)子节点不为空,则将左(右)子节点加入队列 queue ;
将当前层结果 tmp 添加入 res 。
返回值: 返回打印结果列表 res 即可。
原链接:https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof/solution/mian-shi-ti-32-ii-cong-shang-dao-xia-da-yin-er-c-5/
- 代码实现:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> myv;
queue<TreeNode*> q;
if(!root)
return myv;
TreeNode * p=root;
int cengshu=0;
q.push(p);
int count;
while(!q.empty())
{
count=q.size();
vector<int> v;
for(int i=0;i<count;i++)
{
TreeNode* t=q.front();
q.pop();
v.push_back(t->val);
if(t->left)
q.push(t->left);
if(t->right)
q.push(t->right);
}
myv.push_back(v);
}
return myv;
}