第一種也是最常用的一種,使用queue。還有一種不使用queue的方法。不使用queue的思路,其實就是每次都只存儲一層的節點,然后遍歷這一層的節點,是真正的按層遍歷的思想。每次遍歷的都是當前層,記錄的都是當前層的下一層。
public class test
{
public static void main(String[] args)
{
TreeNode root = new TreeNode(1);
TreeNode r1 = new TreeNode(2);
TreeNode r2 = new TreeNode(3);
root.left = r1;
root.right = r2;
test.levelTraversal2(root);
}
//不使用隊列
public static void levelTraversal(TreeNode root)
{
ArrayList<TreeNode> list = new ArrayList<>();
list.add(root);
while(!list.isEmpty())
{
ArrayList<TreeNode> temp = new ArrayList<>();
for (TreeNode node : list)
{
System.out.print(node.val);
if(node.left != null)
{
temp.add(node.left);
}
if(node.right != null)
{
temp.add(node.right);
}
}
list = temp;
}
}
//使用隊列
public static void levelTraversal2(TreeNode root)
{
ArrayDeque<TreeNode> queue = new ArrayDeque<>();
queue.add(root);
while(!queue.isEmpty())
{
TreeNode temp = queue.poll();
System.out.print(temp.val);
if(temp.left != null)
{
queue.offer(temp.left);
}
if(temp.right != null)
{
queue.offer(temp.right);
}
}
}
}
