打印二叉树(Java实现)

22 篇文章 0 订阅
4 篇文章 0 订阅

从上往下打印出二叉树的每个节点,同层节点从左至右打印。

 

思路: 层次遍历; 

import java.util.ArrayList;
import java.util.Queue;       // 此处增加了一个Queue 
import java.util.LinkedList; // 此处增加一个 LinkedList

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/


public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {

        //创建一个类型为TreeNode 的队列, 另外创建一个类型为Integer的ArrayList用于存放结果
        Queue<TreeNode> q = new LinkedList <> ();
        ArrayList<Integer> res = new ArrayList<> ();

        if(root == null) return res; // 注意此处的res是一个空数组

        q.offer(root); // q.offer() 是队列的一个方法,表示向队列尾部添加元素
        

        while(! q.isEmpty()) {
            TreeNode cur = q.poll(); // q.poll()表示从队列头部删除数据
            res.add(cur.val);        // q.add() 也是队列的一个方法,表示向队列尾部添加元素
            
            if(cur.left != null) {q.offer(cur.left);}
            if(cur.right != null) {q.offer(cur.right);}
            
            
        }
        return res;
    }
}

 

引用:

Queue的add()和offer()及remove()和poll()的区别

https://blog.csdn.net/qq_44837912/article/details/104165789

一. add()方法和offer()方法
二者均是向队列尾部添加元素;
当超出队列界限的时,add()方法会抛出异常,offer()方法会返回false。


二. remove()方法和poll()方法
二者均是从队列头部删除元素;
当超出队列界限的时,remove()方法会抛出异常,poll()方法会返回null。
 

 

如下的代码是用了  Deque双端队列 (源于 https://blog.nowcoder.net/n/5360d81e35ad40ae85379261992b4049

import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        Deque<TreeNode> deque = new LinkedList<>();
        ArrayList<Integer> res = new ArrayList<>();

        if(root == null)
            return res;

        deque.add(root);

        while(!deque.isEmpty()){
            TreeNode node = deque.getFirst();
            deque.pollFirst();
            res.add(node.val);

            if(node.left != null)
                deque.addLast(node.left);

            if(node.right != null)
                deque.addLast(node.right);
        }

        return res;
    }
}

 

引用:

Queue和Deque的区别?

https://blog.csdn.net/uz31415926/article/details/99118686

Queue是队列,Deque是双端队列。

public interface Deque<E> extends Queue<E> {
 

方法上的区别如下:

QueueDeque
addaddFirst
offerofferFirst
removeremoveFirst
pollpollFirst
elementgetFirst
peekpeekFirst

 

Java Deque 与 Queue

 

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值