关于算法-数据结构:LeetCode102二叉树的层序遍历

109次阅读

共计 1492 个字符,预计需要花费 4 分钟才能阅读完成。

二叉树的层序遍历

题目形容:给你一个二叉树,请你返回其按 层序遍历 失去的节点值。(即逐层地,从左到右拜访所有节点)。

示例阐明请见 LeetCode 官网。

起源:力扣(LeetCode)
链接:https://leetcode-cn.com/probl…
著作权归领扣网络所有。商业转载请分割官网受权,非商业转载请注明出处。

解法一:利用队列遍历二叉树

队列的特点是先进先出,所以利用队列来遍历二叉树实现层序遍历,具体过程如下:

  • 首先,如果 root 为空,间接返回空的 List。
  • 如果 root 不为空,申明一个队列 nodes,将 root 节点退出到队列中,申明一个 result,而后遍历 nodes 中的节点,遍历过程如下:

    • 首先,用 count 记录以后队列中节点的数量,即为以后层节点的数量,vals 记录以后节点的值;
    • 从 nodes 中按程序取出 count 个节点,并且将相应的节点值放入 vals 中,同时如果以后节点的左右子节点不为空,按从左到右的程序放入 nodes 中。而后将 vals 退出到后果 result 中。
  • 反复遍历 nodes 中的节点,直到 nodes 为空。
  • 最初返回 result 即为层序遍历的后果。
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

public class LeetCode_102 {
    /**
     * 利用队列遍历二叉树:队列的个性是先进先出
     * @param root
     * @return
     */
    public static List<List<Integer>> levelOrder(TreeNode root) {if (root == null) {return new ArrayList<>();
        }
        List<List<Integer>> result = new ArrayList<>();
        // 暂存每一行的节点
        Queue<TreeNode> nodes = new LinkedList<>();
        nodes.add(root);
        while (!nodes.isEmpty()) {
            // 每次遍历一行数据
            List<Integer> vals = new ArrayList<>();
            int count = nodes.size();
            while (count > 0) {TreeNode curNode = nodes.poll();
                vals.add(curNode.val);
                // 将以后节点的左右子节点从左到右放入队列
                if (curNode.left != null) {nodes.add(curNode.left);
                }
                if (curNode.right != null) {nodes.add(curNode.right);
                }
                count--;
            }
            result.add(vals);
        }
        return result;
    }

    public static void main(String[] args) {TreeNode root = new TreeNode(3);
        root.left = new TreeNode(9);
        root.right = new TreeNode(20);
        root.right.left = new TreeNode(15);
        root.right.right = new TreeNode(7);

        for (List<Integer> integers : levelOrder(root)) {for (Integer integer : integers) {System.out.print(integer + " ");
            }
            System.out.println();}
    }
}

【每日寄语】 虚心使人提高,自豪使人落后。

正文完
 0