C++實(shí)現(xiàn)LeetCode(102.二叉樹層序遍歷)
[LeetCode] 102. Binary Tree Level Order Traversal 二叉樹層序遍歷
Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).
For example:
Given binary tree {3,9,20,#,#,15,7},
3
/ \
9 20
/ \
15 7
return its level order traversal as:
[
[3],
[9,20],
[15,7]
]
層序遍歷二叉樹是典型的廣度優(yōu)先搜索 BFS 的應(yīng)用,但是這里稍微復(fù)雜一點(diǎn)的是,要把各個(gè)層的數(shù)分開,存到一個(gè)二維向量里面,大體思路還是基本相同的,建立一個(gè) queue,然后先把根節(jié)點(diǎn)放進(jìn)去,這時(shí)候找根節(jié)點(diǎn)的左右兩個(gè)子節(jié)點(diǎn),這時(shí)候去掉根節(jié)點(diǎn),此時(shí) queue 里的元素就是下一層的所有節(jié)點(diǎn),用一個(gè) for 循環(huán)遍歷它們,然后存到一個(gè)一維向量里,遍歷完之后再把這個(gè)一維向量存到二維向量里,以此類推,可以完成層序遍歷,參見代碼如下:
解法一:
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
if (!root) return {};
vector<vector<int>> res;
queue<TreeNode*> q{{root}};
while (!q.empty()) {
vector<int> oneLevel;
for (int i = q.size(); i > 0; --i) {
TreeNode *t = q.front(); q.pop();
oneLevel.push_back(t->val);
if (t->left) q.push(t->left);
if (t->right) q.push(t->right);
}
res.push_back(oneLevel);
}
return res;
}
};
下面來看遞歸的寫法,核心就在于需要一個(gè)二維數(shù)組,和一個(gè)變量 level,關(guān)于 level 的作用可以參見博主的另一篇博客 Binary Tree Level Order Traversal II 中的講解,參見代碼如下:
解法二:
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> res;
levelorder(root, 0, res);
return res;
}
void levelorder(TreeNode* node, int level, vector<vector<int>>& res) {
if (!node) return;
if (res.size() == level) res.push_back({});
res[level].push_back(node->val);
if (node->left) levelorder(node->left, level + 1, res);
if (node->right) levelorder(node->right, level + 1, res);
}
};
Github 同步地址:
https://github.com/grandyang/leetcode/issues/102
類似題目:
Binary Tree Level Order Traversal II
Binary Tree Zigzag Level Order Traversal
Binary Tree Vertical Order Traversal
Average of Levels in Binary Tree
N-ary Tree Level Order Traversal
參考資料:
https://leetcode.com/problems/binary-tree-level-order-traversal/
到此這篇關(guān)于C++實(shí)現(xiàn)LeetCode(102.二叉樹層序遍歷)的文章就介紹到這了,更多相關(guān)C++實(shí)現(xiàn)二叉樹層序遍歷內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
c++基礎(chǔ)語法:構(gòu)造函數(shù)初始化列表
構(gòu)造函數(shù)需要初始化的數(shù)據(jù)成員,不論是否顯示的出現(xiàn)在構(gòu)造函數(shù)的成員初始化列表中,都會(huì)在該處完成初始化,并且初始化的順序和其在聲明時(shí)的順序是一致的,與列表的先后順序無關(guān)2013-09-09
C++實(shí)現(xiàn)打印1到最大的n位數(shù)
這篇文章主要介紹了C++實(shí)現(xiàn)打印1到最大的n位數(shù),并分析了實(shí)現(xiàn)代碼中語句的跳轉(zhuǎn)技巧,需要的朋友可以參考下2014-09-09
Visual Studio 2019安裝、測試創(chuàng)建c語言項(xiàng)目(圖文教程)
這篇文章主要介紹了Visual Studio 2019安裝、測試創(chuàng)建c語言項(xiàng)目,Visual Studio 2019是完全免費(fèi)的,而且安裝比較簡單,現(xiàn)在把安裝步驟分享給大家,也給大家做個(gè)參考。一起跟隨小編過來看看吧2020-03-03

