题目:
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] ]题解:
两个Queue轮流。
C++版:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> results;
if(root == NULL)
return results;
queue<TreeNode*> q1, q2;
q1.push(root);
while(!q1.empty()) {
vector<int> result;
while(!q1.empty()) {
TreeNode* cur = q1.front();
result.push_back(cur->val);
q1.pop();
if(cur->left != NULL)
q2.push(cur->left);
if(cur->right != NULL)
q2.push(cur->right);
}
results.push_back(result);
q1 = q2;
queue<TreeNode*> empty;
swap(q2, empty);
}
return results;
}
};
Java版:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> results = new ArrayList<List<Integer>>();
if(root == null)
return results;
Queue<TreeNode> q1 = new LinkedList<>();
Queue<TreeNode> q2 = new LinkedList<>();
q1.add(root);
while(q1.size() != 0) {
List<Integer> result = new ArrayList<>();
while(q1.size() != 0) {
TreeNode cur = q1.poll();
result.add(cur.val);
if(cur.left != null)
q2.add(cur.left);
if(cur.right != null)
q2.add(cur.right);
}
results.add(result);
q1.addAll(q2);
q2.clear();
}
return results;
}
}
Python版:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
import copy
class Solution(object):
def levelOrder(self, root):
"""
:type root: TreeNode
:rtype: List[List[int]]
"""
if root == None:
return []
results = []
q1, q2 = [], []
q1.append(root)
while len(q1) != 0:
result = []
for i in q1:
result.append(i.val)
if i.left != None:
q2.append(i.left)
if i.right != None:
q2.append(i.right)
results.append(result)
q1 = copy.copy(q2)
q2 = []
return results