107. 二叉樹的層次遍歷 II
阿新 • • 發佈:2021-06-23
難度 medium
給定一個二叉樹,返回其節點值自底向上的層次遍歷。 (即按從葉子節點所在層到根節點所在的層,逐層從左向右遍歷)
例如:
給定二叉樹 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其自底向上的層次遍歷為:
[
[15,7],
[9,20],
[3]
]
解題思路:用單向佇列來層序遍歷二叉樹是通用的做法,這裡因為是從下到上遍歷,每次將一層的資料儲存在一個一維陣列中,插入二維陣列的時候保證插入的位置是第0位即可,這樣就能保證結果是逆序的。
程式碼 t100 s14 java
class Solution { public List<List<Integer>> levelOrderBottom(TreeNode root) { List<List<Integer>> res = new ArrayList<>(); Queue<TreeNode> pq = new LinkedList<>(); if(root==null) return res; pq.add(root); while(!pq.isEmpty()){ int sz = pq.size(); List<Integer> t = new ArrayList<>(); for(int i=0; i<sz; i++){ TreeNode node = pq.poll(); t.add(node.val); if(node.left!=null) pq.add(node.left); if(node.right!=null) pq.add(node.right); } res.add(0, t); } return res; } }
下面這個是4個月前的cpp版本,用一個層次遍歷,先按從上往下的順序把每一層push進一個vector< vector< int>>裡面,然後再反轉一下。
程式碼t53 s8 cpp
class Solution { public: vector<vector<int>> levelOrderBottom(TreeNode* root) { vector<vector<int>> res; if(!root) return res; queue<TreeNode*> q; q.push(root); while(!q.empty()){ int t = q.size(); vector<int> vec; for(int i=0; i<t; i++){ TreeNode* temp = q.front(); q.pop(); vec.push_back(temp->val); if(temp->left) q.push(temp->left); if(temp->right) q.push(temp->right); } res.push_back(vec); } reverse(res.begin(), res.end()); return res; } };
參考資料: