leetcode-每个节点的右向指针(填充同一层的兄弟节点)

给定一个二叉树

struct TreeLinkNode {
  TreeLinkNode *left;
  TreeLinkNode *right;
  TreeLinkNode *next;
}

填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL

初始状态下,所有 next 指针都被设置为 NULL

说明:

  • 你只能使用额外常数空间。
  • 使用递归解题也符合要求,本题中递归程序占用的栈空间不算做额外的空间复杂度。
  • 你可以假设它是一个完美二叉树(即所有叶子节点都在同一层,每个父节点都有两个子节点)。

示例:

给定完美二叉树,

     1
   /  \
  2    3
 / \  / \
4  5  6  7

调用你的函数后,该完美二叉树变为:

     1 -> NULL
   /  \
  2 -> 3 -> NULL
 / \  / \
4->5->6->7 -> NULL

递归法:因为是完美二叉树,如果有左子树,一定有右子树。我们将结点的左子树和右子树相连。
如果结点next不为空,那么将结点右子树指向结点next的左子树。
同时递归处理左子树和右子树。
public class Solution {
    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);
    }
}

 

层序遍历法:

/*
*利用层序遍历,每一层都将指针指向他的下一个结点,
*这里用size标记每一层是否遍历完毕。
*
*/
public class Solution {
    public void connect(TreeLinkNode root) {
       if(root==null)return ;
        Queue<TreeLinkNode> q=new LinkedList();
        q.add(root);
        while(!q.isEmpty()){
            int size=q.size();
            for(int i=0;i<size;i++){
                TreeLinkNode temp=q.peek();q.poll();
                if(i<size-1)temp.next=q.peek();
                if(temp.left!=null)q.add(temp.left);
                if(temp.right!=null)q.add(temp.right);
            }
        }
    }
}

 

最巧妙地:利用层指针和深度的指针

public class Solution {
    public void connect(TreeLinkNode root) {
        if(root==null)return ;
        TreeLinkNode cur=null,start=root;
        while(start.left!=null){
            cur=start;
            while(cur!=null){
                cur.left.next=cur.right;
               if(cur.next!=null)cur.right.next=cur.next.left;
                cur=cur.next;
            }
            start=start.left;           //这个地方是left,不是next
        }
    }
}
posted @ 2018-08-26 19:10  吃土豆鸭  阅读(998)  评论(0编辑  收藏  举报