[leetcode]Populating Next Right Pointers in Each Node - java 广度遍历

最新推荐文章于 2024-09-14 18:52:56 发布

iteye_9083

最新推荐文章于 2024-09-14 18:52:56 发布

阅读量67

点赞数

分类专栏： leetcode 文章标签： leetcode 算法

leetcode 专栏收录该内容

44 篇文章 0 订阅

订阅专栏

这题没啥难度，主要是仔细了

/**

 * Definition for binary tree with next pointer.

 * public class TreeLinkNode {

 *     int val;

 *     TreeLinkNode left, right, next;

 *     TreeLinkNode(int x) { val = x; }

 * }

 */

public class Solution {

   public void connect(TreeLinkNode root) {

        if(root == null){

            return;

        }

        int k = 0;

        int length = (int)Math.pow(2, k);

        LinkedList<TreeLinkNode> queue = new LinkedList<TreeLinkNode>();

        queue.add(root);

        TreeLinkNode preNode = null;

        while (!queue.isEmpty()){

            TreeLinkNode node = queue.remove();

            if(node.left != null){

                queue.add(node.left);

            }

            if(node.right!=null){

                queue.add(node.right);

            }

            if(preNode != null){

                preNode.next=node;

            }

            if(length == 1){

                length = (int)Math.pow(2, ++k);

                node.next=null;

                preNode = null;

            }else{

                preNode = node;

                length--;

            }

        }

    }

}