binary tree level order traversal ii)(二叉樹自下向上層序遍歷)
題目描述
Given a binary tree, return the bottom-up level order traversal of its nodes’ values. (ie, from left to right, level by level from leaf to root).
For example:
Given binary tree{3,9,20,#,#,15,7},
return its bottom-up level order traversal as:
confused what"{1,#,2,3}"means? > read more on how binary tree is serialized on OJ.
OJ’s Binary Tree Serialization:
The serialization of a binary tree follows a level order traversal, where ‘#’ signifies a path terminator where no node exists below.
Here’s an example:
The above binary tree is serialized as"{1,2,3,#,#,4,#,#,5}".
題目大意
實現二叉樹自底層向上層的層序遍歷。
思路
還是二叉樹層序遍歷的問題,只不過是自下向上;
很好解決
在C++中,可以用vector,可以實現在vector前邊插入:
vector<vector<int > > vec; // 定義二維陣列,其中元素為int型別
vector<int > vt; // 二維陣列的新一行
vec.insert(vec.begin(), vt); // 在二維陣列前面插入新的一行
或者在Java中:
ArrayList<ArrayList<Integer>> res = new ArrayList<>(); ArrayList<Integer> list = new ArrayList<>(); res.add(0, list);
也可以實現在二維陣列前面插入一行。
但是經過我在用C++的實驗,發現先用vec.push_back(vt)的方式,插入,然後最後的時候用swap(vec[i], vec[j])交換一下,不管是空間還是時間,效率更優。
#include<iostream>
#include<vector>
#include<queue>
using namespace std;
/*
結構體定義
*/
struct TreeNode
{
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
/*
具體實現演算法
*/
typedef TreeNode* tree;
vector<vector<int> > levelOrderBottom(TreeNode *root)
{
vector<vector<int > > vec; // 定義二維陣列,其中元素為int型別
if(root == NULL)return vec;
queue<tree> qu; // 儲存二叉樹層序遍歷結點的指標
qu.push(root); // 頭指標入隊
while(!qu.empty())
{
int index = qu.size(); // 本層的結點個數
vector<int > vt; // 二維陣列的新一行
tree now; // 暫存當前結點
while(index--)
{
now = qu.front(); // 暫存當前結點
qu.pop(); // 出隊
vt.push_back(now->val);
if(now->left != NULL)qu.push(now->left); // 入隊
if(now->right != NULL)qu.push(now->right); // 入隊
}
// 如果vt不為空,則加入到二維陣列的新一行中
// 其實分析可以發現,vt也不可能為空
if(vt.size())
vec.push_back(vt);
}
// 因為自下向上,所以換一下
for(int i=0,j=vec.size()-1; i<j; i++,j--)
{
swap(vec[i], vec[j]);
}
return vec;
}
// 二叉樹的層序遍歷演算法
void print(TreeNode *root)
{
queue<tree > qu;
qu.push(root);
while(!qu.empty())
{
tree now = qu.front();
qu.pop();
cout<<now->val<<endl;
if(now->left != NULL)qu.push(now->left);
if(now->right != NULL)qu.push(now->right);
}
}
int main()
{
tree tr;
tr = new TreeNode(1);
tree t1;
t1 = new TreeNode(2);
tr->left = t1;
tree t2;
t2 = new TreeNode(3);
tr->right = t2;
tree t3;
t3 = new TreeNode(4);
t2->left = t3;
vector<vector<int > > vec;
//print(tr);
vec = levelOrderBottom(tr);
for(int i=0; i<vec.size(); i++)
{
for(int j=0; j<vec[i].size(); j++)
{
cout<<vec[i][j]<<' ';
}
cout<<endl;
}
return 0;
}
以上。