Leetcode 102. Binary Tree Level Order Traversal

102. Binary Tree Level Order Traversal

Total Accepted: 110104 Total Submissions: 327888 Difficulty: Easy

 

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,null,null,15,7],

   3
   / \
  9  20
    /  \
   15   7

return its level order traversal as:

[
  [3],
  [9,20],
  [15,7]
]

 

思路：由顶至下的层序遍历，输出每一层不为空节点的值。

代码：

1.BFS

代码中遍历一层时，用了3个指针：begin，end，cur，分别记录当前层的第1个非空节点，当前层最后1个非空节点，下一层最后1个非空节点。用BFS遍历。

 1 /**
 2  * Definition for a binary tree node.
 3  * struct TreeNode {
 4  *     int val;
 5  *     TreeNode *left;
 6  *     TreeNode *right;
 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 8  * };
 9  */
10 class Solution {
11 public:
12     vector<vector<int>> levelOrder(TreeNode* root) {
13         vector<vector<int>> res;
14         if(root==NULL) return res;
15         queue<TreeNode*> q;
16         q.push(root);
17         TreeNode *begin,*end=root,*cur=root;
18         while(!q.empty()){
19             vector<int> temp;
20             begin=q.front();
21             q.pop();
22             while(begin!=end){
23                 temp.push_back(begin->val);
24                 if(begin->left){
25                     q.push(begin->left);
26                     cur=begin->left;
27                 }
28                 if(begin->right){
29                     q.push(begin->right);
30                     cur=begin->right;
31                 }
32                 begin=q.front();
33                 q.pop();
34             }
35             temp.push_back(begin->val);
36             if(begin->left){
37                 q.push(begin->left);
38                 cur=begin->left;
39             }
40             if(begin->right){
41                 q.push(begin->right);
42                 cur=begin->right;
43             }
44             res.push_back(temp);
45             end=cur;
46         }
47         return res;
48     }
49 };

 

类似的，作为一个技巧，可以先插入NULL到队列中，作为分层的标记：

 1 class Solution {
 2 public:
 3     vector<vector<int> > levelOrder(TreeNode* root) {
 4         vector<vector<int> > res;
 5         if(root==NULL) return res;
 6         vector<int> temp;
 7         queue<TreeNode*> q;
 8         q.push(root);
 9         q.push(NULL);
10         TreeNode* cur;
11         while(!q.empty()){
12             cur=q.front();
13             q.pop();
14             while(cur){
15                 temp.push_back(cur->val);
16                 if(cur->left) q.push(cur->left);
17                 if(cur->right) q.push(cur->right);
18                 cur=q.front();
19                 q.pop();
20             }
21             result.push_back(temp);
22             if(q.size()>0){
23                 temp.resize(0);
24                 q.push(NULL);
25             }
26         }
27         return res;
28     }
29 };

 

2.DFS

 1 class Solution {
 2 public:
 3     vector<vector<int> > res;
 4 
 5     void DFS(TreeNode* root, int level)
 6     {
 7         if(root==NULL) return;
 8         if(res.size()==level){//一开始
 9             res.push_back(vector<int>());
10         }
11         res[level].push_back(root->val);
12         DFS(root->left,level+1);
13         DFS(root->right,level+1);
14     }
15     
16     vector<vector<int> > levelOrder(TreeNode *root) {
17         DFS(root, 0);
18         return res;
19     }
20 };

 

 

姐妹题：107. Binary Tree Level Order Traversal II

做法可以相同，就是返回的结果反个顺序。

这里注意一下：res是vector容器，则有

res.begin() 返回一个迭代器，它指向容器c的第一个元素

res.end() 返回一个迭代器，它指向容器c的最后一个元素的下一个位置

res.rbegin() 返回一个逆序迭代器，它指向容器c的最后一个元素

res.rend() 返回一个逆序迭代器，它指向容器c的第一个元素前面的位置

posted @ 2016-06-29 12:51  Deribs4  阅读(168)  评论(0)  编辑  收藏  举报