LeetCode 107 ——二叉樹的層次遍歷 II
阿新 • • 發佈:2018-11-17
1. 題目
2. 解答
與 LeetCode 102 ——二叉樹的層次遍歷 類似,我們只需要將每一層的資料倒序輸出即可。
定義一個存放樹中資料的向量 data,一個存放樹的每一層資料的向量 level_data 和一個存放每一層節點的佇列 node_queue。
如果根節點非空,根節點進隊,然後迴圈以下過程直至佇列為空:
-
- 得到佇列的大小,即為樹中當前層的節點個數。佇列元素迴圈出隊,並將節點的值加入 level_data,如果節點有左右子節點,左右子節點入隊
-
- 將 level_data 插入到 data 頭部並清空 level_data
/** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */ class Solution { public: vector<vector<int>> levelOrder(TreeNode* root) { vector<vector<int>> data; vector<int> level_data; queue<TreeNode *> node_queue; int node_num = 1; TreeNode* temp = NULL; if (root) node_queue.push(root); while (!node_queue.empty()) { node_num = node_queue.size(); for (int i = 0; i < node_num; i++) { temp = node_queue.front(); node_queue.pop(); level_data.push_back(temp->val); if (temp->left) node_queue.push(temp->left); if (temp->right) node_queue.push(temp->right); } data.push_back(level_data); level_data.clear(); } return data; } };
獲取更多精彩,請關注「seniusen」!