- 题目描述:
- 设计思想:
详细操作流程看下图
-视频讲解链接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 };