LeetCode 102.二叉树的层序遍历 C++
思路🧐:
用队列存储结点,vector存储每层结点的值,再用vector<vector<int>>存储整个二叉树的值,要获取每层结点个数,可以创建一个变量levelsize,读取队列所存储的结点个数。
代码🔎:
class Solution { public: vector<vector<int>> levelOrder(TreeNode* root) { queue<TreeNode*> q; if(root) //root不为空 { q.push(root); } vector<vector<int>> vv; int levelSize = 1; //root最开始只有一个结点 while(!q.empty()) { vector<int> v; //存储每个结点的值 while(levelSize--) //一层数据走完就结束循环 { 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); } levelSize = q.size(); //更新每层结点个数 vv.push_back(v); //存储每层结点的值 } return vv; } };