广告公司网站开发,wordpress点击阅读全部,海南注册公司税收优惠政策,备案号 网站题目描述
给你二叉树的根节点 root #xff0c;返回其节点值的 层序遍历 。 #xff08;即逐层地#xff0c;从左到右访问所有节点#xff09;。
示例1#xff1a; 输入#xff1a;root [3,9,20,null,null,15,7]
输出#xff1a;[[3],[9,20],[15,7]]示例 2#xff1…题目描述
给你二叉树的根节点 root 返回其节点值的 层序遍历 。 即逐层地从左到右访问所有节点。
示例1 输入root [3,9,20,null,null,15,7]
输出[[3],[9,20],[15,7]]示例 2
输入root [1]
输出[[1]]示例 3
输入root []
输出[]提示
树中节点数目在范围 [0, 2000] 内-1000 Node.val 1000
思路分析
这个问题实际上可以只用一个队列就实现只需要再增加一个变量levelSize用来记录每一层的数据个数然后再让这个队列一层一层的出去。之前的方法中实际上队列并不是一层一层出去的它有可能队列里面同时有两层的数据我们以下面这个图来解释一下原因 如果有两层队列实现的话3这个节点出来的时候会让9和20这两个节点进入队列而9这个节点出来的时候会让15这个节点进入队列这个时候队列里面就同时有了第2层和第3层的数据。
所以我们想通过levelSize来达到一个目的控制这个队列实现一层一层的出去。
那我们要怎么实现呢我们仍然以刚才的图来进行分析
3节点进入队列的时候它的层数为1由于它是根节点所以它肯定只有一个所以3就可以直接出队列。这个时候我们让levelSize进行自减操作它就变成了0表示这一层已经出完了。
那么由于3节点出的时候会把9和20也带进来也就是说当前层的节点全部出队列的时候一定是下一层的节点全部进入队列这个时候我们将levelSize重新更新为第二层节点的数目也就是2然后再进行出队列的操作9节点出队列同时将15节点带进队列然后levelSize自减变为120节点出队列同时将15节点和7节点带进队列levelSize再自减变为0。这个时候就说明第二层也出完了。那么此时第三层都在队列里面所以我们再次更新levelSize的值为3依次类推直到整棵树都被遍历完就实现了只用一个队列实现层序遍历。
那么根据以上的思路我们就可以写出下面的代码
完整代码
class Solution {
public:vectorvectorint levelOrder(TreeNode* root) {queueTreeNode* q;int levelSize 0;if (root)//如果根不为空就入队列{q.push(root);levelSize 1;}vectorvectorint vv;//用来存放一层一层出的节点while (!q.empty())//如果队列不等于空就说明树还没有被遍历完{//通过levelSize控制一层一层出vectorint v;//用来存放每一层的数据while (levelSize--)//levelSize是几循环就执行几次--levelSize表示的则是执行levelSize - 1次{TreeNode* front q.front();//先取队头的数据q.pop();v.push_back(front-val);//进去的同时把该节点的下一层往队列里面带if (front-left)//左如果不为空就让左入队列q.push(front-left);if (front-right)//右如果不为空就让右入队列q.push(front-right);}//走到这里就说明当前层已经出完了就把当前层所出的数据放到vv里面vv.push_back(v);//更新下一层的数据levelSize q.size();}return vv;}
};运行结果
