-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathLeetCode_102.cpp
49 lines (48 loc) · 1.46 KB
/
LeetCode_102.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
/*************************************************************************
> File Name: LeetCode_102.cpp
> Author:
> Mail:
> Created Time: Tue 07 Apr 2020 08:12:19 PM PDT
************************************************************************/
#include<iostream>
#include<vector>
using namespace std;
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> levelTra;
if (root == nullptr)
return levelTra;
queue<TreeNode*> treeQue;
treeQue.push(root);
while (!treeQue.empty())
{
vector<int> curLevl;
int size = treeQue.size();
for (int i = 0; i < size; i++)
{
TreeNode* pNode = treeQue.front();
treeQue.pop();
curLevl.push_back(pNode->val);
if (pNode->left != nullptr)
treeQue.push(pNode->left);
if (pNode->right != nullptr)
treeQue.push(pNode->right);
}
levelTra.emplace_back(curLevl);
}
return levelTra;
}
void LevelCore(TreeNode* root, vector<vector<int>>& result, int curLevel)
{
if (root == nullptr)
return;
if (curLevel >= result.size())
result.push_back(vector<int>());
result[curLevel].push_back(root->val);
LevelCore(root->left, result, curLevel + 1);
LevelCore(root->right, result, curLevel + 1);
}
vector<vector<int>> levelOrderRecursively(TreeNode* root){
vector<vector<int>> result;
LevelCore(root, result, 0);
return result;
}