刷題-力扣-102. 二叉樹的層序遍歷
阿新 • • 發佈:2021-06-17
102. 二叉樹的層序遍歷
題目連結
來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/binary-tree-level-order-traversal/
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。
題目描述
給你一個二叉樹,請你返回其按 層序遍歷 得到的節點值。 (即逐層地,從左到右訪問所有節點)。
示例:
二叉樹:[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其層序遍歷結果:
[
[3],
[9,20],
[15,7]
]
題目分析
- 根據題目描述層次遍歷二叉樹
- 使用佇列FIFO,父節點出棧子節點進棧,每次記錄需要出棧的幾點個數,即二叉樹中該層的節點數
程式碼
/** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode() : val(0), left(nullptr), right(nullptr) {} * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {} * }; */ class Solution { public: vector<vector<int>> levelOrder(TreeNode* root) { vector<vector<int>> res; if (root == nullptr) return res; queue<TreeNode*> node; node.push(root); while (!node.empty()) { int nodeLen = node.size(); res.push_back(vector<int>()); while (nodeLen--) { if (node.front()->left) node.push(node.front()->left); if (node.front()->right) node.push(node.front()->right); res.back().push_back(node.front()->val); node.pop(); } } return res; } };