这是一题关于树的遍历的题目,解题思路是在二叉树的层次遍历算法中稍作修改,让二叉树的每一层都能存储在不同的集合中。
以下是解题代码,所有的解题思路都写在代码中
using System.Collections.Generic;
/*
public class TreeNode
{
public int val;
public TreeNode left;
public TreeNode right;
public TreeNode (int x)
{
val = x;
}
}*/
/*从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。*/
class Solution
{
public List<List<int>> Print(TreeNode pRoot)
{
List<List<int>> treeList=new List<List<int>>();
Queue<TreeNode> queue=new Queue<TreeNode>();
if(pRoot==null) //这段是为了应付测试用例的
{
return treeList;
}
queue.Enqueue(pRoot);//先把最开始的根节点入队
while(queue.Count!=0)
{
List<int> row=new List<int>();//这是存储每一层节点的集合
int count=queue.Count;//获取当前集合内节点的数量
while(count>0)//当前层节点的数量不为零,这次循环就不会停止
{
TreeNode temp=queue.Peek();//获取队列的头节点,存储在临时节点temp中
queue.Dequeue();//头节点出队
if(temp.left!=null)//如果刚才出队的头节点有左右孩子,将左右孩子放入队列中
{
queue.Enqueue(temp.left);
}
if(temp.right!=null)
{
queue.Enqueue(temp.right);
}
//注意,因为当前层节点的数量已经固定了(count)
//故新加入的这些节点属于下一层不会被这次遍历出队
row.Add(temp.val);//刚才出队的节点放入集合中
count--;//当前层的节点数量减1
}//循环结束,说明当前层的节点值都全部已经被加入Lisr集合中了
treeList.Add(row);//把这一层的List放入treeList中
//开始下一层的遍历
}
return treeList;
}
}