242 · 将二叉树按照层级转化为链表

最新推荐文章于 2024-09-04 09:10:25 发布

阿飞算法

最新推荐文章于 2024-09-04 09:10:25 发布

阅读量79

点赞数

分类专栏： Lintcode

本文链接：https://blog.csdn.net/wat1r/article/details/118598757

版权

Lintcode 专栏收录该内容

31 篇文章 0 订阅

订阅专栏

242 · 将二叉树按照层级转化为链表

方法1：BFS层序遍历

        public List<ListNode> binaryTreeToLists(TreeNode root) {
            List<ListNode> res = new ArrayList<>();
            if (root == null) return res;
            Queue<TreeNode> q = new LinkedList<>();
            q.offer(root);
            while (!q.isEmpty()) {
                ListNode head = null;
                ListNode cur = null;
                int size = q.size();
                for (int i = 0; i < size; i++) {
                    TreeNode node = q.poll();
                    if (i == 0) {
                        head = new ListNode(node.val);
                        cur = head;
                    } else {
                        cur.next = new ListNode(node.val);
                        cur = cur.next;
                    }
                    if (node.left != null) q.offer(node.left);
                    if (node.right != null) q.offer(node.right);
                }
                res.add(head);
            }
            return res;
        }