- 题目描述:
- 设计思想:
详细操作流程看下图
-视频讲解链接B站视频讲解
- 代码:
c++版本:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
class Solution {
public:
vector<vector<int> > Print(TreeNode* root) {
vector<vector<int> >res; //用于存储返回结果
queue<TreeNode*> q; //创建队列用于存储节点
if(root == NULL) return res; //当为空的时候直接返回
int height = 1; //用于处理之字形遍历,偶数就直接插入,偶数就插入temp头
q.push(root); //插入节点
while(!q.empty()){
vector<int> temp; // 存储每一层的数字
int n = q.size(); // 队列大小表示当前层数的元素个数
for(int i = 0;i < n;i ++){
TreeNode* node = q.front(); //取出队列的第一个元素
q.pop();
if(height % 2 == 0){
temp.insert(temp.begin(),node->val); // 如果层数是偶数就插入到头
}else{
temp.push_back(node->val); //如果层数是奇数就直接放进去
}
if(node->left != NULL) q.push(node->left);//如果左子树不为空就递归左子树
if(node->right!= NULL) q.push(node->right);//如果右子树不为空就递归右子树
}
height ++; //高度++
res.push_back(temp); //把这一层的节点插入到res中
}
return res;
}
};
Java版本:
import java.util.*;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public ArrayList<ArrayList<Integer> > Print(TreeNode root) {
ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>(); //用于存储返回结果
Queue<TreeNode> q = new LinkedList<TreeNode>();//创建队列用于存储节点
if(root == null){
return res; //当为空的时候直接返回
}
int height = 1;//用于处理之字形遍历,奇数就直接插入,偶数就插入temp头
q.add(root); //插入节点
while(!q.isEmpty()){
ArrayList<Integer> temp = new ArrayList<Integer>();// 存储每一层的数字
int n = q.size();// 队列大小表示当前层数的元素个数
for(int i = 0;i < n;i ++){
TreeNode node = q.poll();//取出队列的第一个元素
if(height % 2 == 0){
temp.add(0,node.val);// 如果层数是偶数就插入到头
}else{
temp.add(node.val); //如果层数是奇数就直接放进去
}
if (node.left != null)
q.add(node.left); //如果左子树不为空就递归左子树
if (node.right != null)
q.add(node.right); //如果右子树不为空就递归右子树
}
height ++; //高度++
res.add(temp); //把这一层的节点插入到res中
}
return res;
}
}
Python版本:
import queue
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
#
#
# @param root TreeNode类
# @return int整型二维数组
#
class Solution:
def Print(self , root ):
# write code here
res = [] #用于返回最后的结果
if not root:
return res #如果根节点为空就返回结果
q = queue.Queue() #创建队列用于存储节点
q.put(root)
height = 1 #用于处理之字形遍历,奇数就直接插入,偶数就插入temp头
while q.qsize() >0:
temp = [] #用于存储当前遍历这一层的节点
n = q.qsize() # 队列大小表示当前层数的元素个数
for i in range(n):
node = q.get()#取出队列的第一个元素
if height%2 == 0:
temp.insert(0,node.val) # 如果层数是偶数就插入到头
else:
temp.append(node.val) #如果层数是奇数就直接放进去
if node.left:
q.put(node.left)#左孩子如果不为空就进队列
if node.right:
q.put(node.right)#右孩子如果不为空就进队列
res.append(temp) #将这一层的节点数里面据保存到res
height += 1 #高度++
return res
JavaScript代码:
/* function TreeNode(x) {
this.val = x;
this.left = null;
this.right = null;
} */
function Print(root)
{
// write code here
let res = []; //用于存储返回结果
let q = [];//创建队列用于存储节点
if(root == null){
return res; //当为空的时候直接返回
}
let height = 1;//用于处理之字形遍历,奇数就直接插入,偶数就插入temp头
q.push(root); //插入节点
while(q.length != 0){
let temp = [];// 存储每一层的数字
let n = q.length;// 队列大小表示当前层数的元素个数
for(let i = 0;i < n;i ++){
let node = q.shift();//取出队列的第一个元素
if(height % 2 == 0){
temp.unshift(node.val);// 如果层数是偶数就插入到头
}else{
temp.push(node.val); //如果层数是奇数就直接放进去
}
if (node.left != null)
q.push(node.left); //如果左子树不为空就递归左子树
if (node.right != null)
q.push(node.right); //如果右子树不为空就递归右子树
}
height ++; //高度++
res.push(temp); //把这一层的节点插入到res中
}
return res;
}
module.exports = {
Print : Print
};
京公网安备 11010502036488号