美文网首页
二叉树的层次遍历

二叉树的层次遍历

作者: 二进制的二哈 | 来源:发表于2019-12-21 15:49 被阅读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);
        }
    }
    

    相关文章

      网友评论

          本文标题:二叉树的层次遍历

          本文链接:https://www.haomeiwen.com/subject/spegnctx.html