用队列来求二叉树的高度、宽度、层序遍历

这里其实用到的是同一个思想,都是从根节点开始入队,依次将左、右子节点出队,外层循环用队列非空进行判断,当为空时,二叉树遍历完;内层循环是循环每一层的节点数,并同时将该层子节点入队。
1、用队列(非递归)求树的高度

public static int TreeDepth2(TreeNode root) {
        if (root==null){
            return 0;
        }

        int height = 0;
        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.add(root);

        while (!queue.isEmpty()){
            int len = queue.size();

            for (int i = 0;i<len;i++){
                TreeNode cur = queue.pollFirst();
                if (cur.left!=null){
                    queue.add(cur.left);
                }
                if (cur.right!=null){
                    queue.add(cur.right);
                }
            }
            height++;
        }

        return height;
    }

2、用队列(非递归)求树的宽度

public static int TreeWidth(TreeNode root){
        if (root==null){
            return 0;
        }

        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        int width = 0;

        while (!queue.isEmpty()){
            int len = queue.size();
            width = Math.max(len, width);

            for (int i = 0;i<len;i++){
                TreeNode cur = queue.pollFirst();
                if (cur.left!=null){
                    queue.add(cur.left);
                }
                if (cur.right!=null){
                    queue.add(cur.right);
                }
            }
        }

        return width;
    }

3、用队列实现二叉树的层序遍历
每层节点放入到List中,最后放入到大List中

public static List<List<Integer>> levelOrder(TreeNode root){
        if (root==null){
            return null;
        }
        List<List<Integer>> res = new ArrayList<>();

        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.add(root);

        while (!queue.isEmpty()){
            int len = queue.size();
            List<Integer> list = new ArrayList<>();
            for (int i = 0;i<len;i++){
                TreeNode cur = queue.pollFirst();
                list.add(cur.val);
                if (cur.left!=null){
                    queue.add(cur.left);
                }
                if (cur.right!=null){
                    queue.add(cur.right);
                }
            }
            res.add(list);
        }

        return res;
    }

所以,这三种问题本质上是一个东西,只是在循环的过程中进行了不同操作,从而实现了求高度、求宽度和层序遍历。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值