116. Populating Next Right Pointers in Each Node

標籤：root   add   connect   分享圖片   image   style   ISE   poll   empty   

一、題目

　　1、審題

　　2、分析

　　　　給出一個完全二叉樹，添加二叉樹的 next 指標指向。

 

二、解答

　　1、思路：　

　　　　方法一、

　　　　　　採用隊列進行層次遍曆，遍曆時添加 next 指標。

    public void connect(TreeLinkNode root) {            if(root == null)            return;        Queue<TreeLinkNode> queue = new LinkedList<TreeLinkNode>();        queue.add(root);        TreeLinkNode node;                while(!queue.isEmpty()) {                        int size = queue.size();            for (int i = 0; i < size; i++) {                node = queue.poll();                if(i < size - 1)                    node.next = queue.peek();                if(node.left != null) {                    queue.add(node.left);                }                if(node.right != null) {                    queue.add(node.right);                }                                }        }    }

　　

　　方法二、

　　　　利用兩個指標進行層次遍曆，添加 next 指標

    public void connect(TreeLinkNode root) {        if(root == null)            return;        TreeLinkNode pre = root;        TreeLinkNode cur;        while(pre.left != null) {            cur = pre;            while(cur != null) {                cur.left.next = cur.right;                if(cur.next != null)                    cur.right.next = cur.next.left;                cur = cur.next;            }            pre = pre.left;        }    }

　　

　　方法三、

　　　　利用遞迴實現每一層的 next 指標。

    public void connect(TreeLinkNode root) {        if(root == null)            return;                if(root.left != null) {            root.left.next = root.right;            if(root.next != null)                root.right.next = root.next.left;        }        connect(root.left);        connect(root.right);    }

 

116. Populating Next Right Pointers in Each Node