Java

二叉树的层序遍历

2019-10-16  本文已影响0人  JAVA觅音阁

二叉树的层序遍历,分为自上而下和自下而上进行层序遍历。

  • 自上而下:从根结点层往下逐层从左往右访问。
  • 自下而上:从叶子结点层往上逐层从左往右访问。

1.自上而下层序遍历

思路:利用队列先进先出的特性。

1.创建Queue,根节点入队;
2.队列不为空,记录此时的队列长度(即该层的元素个数);
3.按以上长度循环遍历队列,取队列元素,将结果存到一个一维向量,然后入队其左右两个子节点。将结果添加到二维向量,重复步骤2。

JAVA参考代码

public List<List<Integer>> levelOrderFromTop(TreeNode root) {
    List<List<Integer>> result = new ArrayList();

    if (root == null) {
        return result;
    }

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

    while (!queue.isEmpty()) {
        List<Integer> level = new ArrayList<>();
        int size = queue.size();
        for (int i = 0; i < size; i++) {
            TreeNode head = queue.poll();
            level.add(head.val);
            if (head.left != null) {
                queue.offer(head.left);
            }
            if (head.right != null) {
                queue.offer(head.right);
            }
        }
        result.add(level);
    }

    return result;
}

2.自下而上层序遍历

思路:利用队列先进先出的特性。

.1创建Queue,根节点入队;
2.队列不为空,记录此时的队列长度(即该层的元素个数);
3.按以上长度循环遍历队列,取队列元素,将结果存到一个一维向量,然后入队其左右两个子节点。将结果添加到二维向量,重复步骤2;
4.以上结果为自上而下的层序遍历,反转二维向量即可得到自下而上的层序遍历结果。

JAVA参考代码

public List<List<Integer>> levelOrderFromBottom(TreeNode root) {
    List<List<Integer>> result = new ArrayList<>();
    if (root == null) {
        return result;
    }
    Queue<TreeNode> queue = new LinkedList<>();
    queue.offer(root);

    while (!queue.isEmpty()) {
        List<Integer> level = new ArrayList<>();
        int size = queue.size();
        for (int i = 0; i < size; i++) {
            TreeNode head = queue.poll();
            level.add(head.val);
            if (head.left != null) {
                queue.offer(head.left);
            }
            if (head.right != null) {
                queue.offer(head.right);
            }
        }
        result.add(level);
    }

    Collections.reverse(result);
    return result;
}
上一篇下一篇

猜你喜欢

热点阅读