二叉树的层次遍历

2019-12-21  本文已影响0人  二进制的二哈

题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-level-order-traversal

给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果:

[
  [3],
  [9,20],
  [15,7]
]

递归解法:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> ans = new ArrayList<>();
        if (root != null){
            List<TreeNode> nodes = new ArrayList<>();
            nodes.add(root);
            func(ans,nodes);
        }
        return ans;
    }

    private void func(List<List<Integer>> ans,List<TreeNode> nodes){
        if (nodes.isEmpty())
            return;
        List<Integer> tmp = new ArrayList<>();
        List<TreeNode> newNodes = new ArrayList<>();
        for (TreeNode node : nodes) {
            tmp.add(node.val);
            if (node.left != null)
                newNodes.add(node.left);
            if (node.right != null)
                newNodes.add(node.right);
        }
        ans.add(tmp);
        func(ans,newNodes);
    }
}
上一篇下一篇

猜你喜欢

热点阅读