leetcode-填充同一层的兄弟节点Ⅱ

  给定一个二叉树
  
  struct TreeLinkNode {
  
  TreeLinkNode *left;
  
  TreeLinkNode *right;
  
  TreeLinkNode *next;
  
  }
  
  填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。
  
  初始状态下,所有 next 指针都被设置为 NULL。
  
  说明:
  
  你只能使用额外常数空间。
  
  使用递归解题也符合要求,本题中递归程序占用的栈空间不算做额外的空间复杂度。
  
  示例:
  
  给定二叉树,
  
  1
  
  /  \
  
  2    3
  
  / \    \
  
  4   5    7
  
  调用你的函数后,该二叉树变为:
  
  1 -> NULL
  
  /  \
  
  2 -> 3 -> NULL
  
  / \    \
  
  4-> 5 -> 7 -> NULL
  
  第一道题是:
  
  我们依然可以使用层序遍历的方法,为每个结点添加next指针。
  
  但是时间复杂度不满足要求。
  
  复制代码
  
  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(www.mingcheng178.com);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);
  
  }
  
  }
  
  }
  
  复制代码
  
  将第一题的递归方法进行改造:
  
  比如该二叉树
  
     1
  
  /  \
  
  2    3
  
  / \    \
  
  4   5    7
  
  \    / \
  
  9  14   15
  
  / \
  
  18  19
  
  遍历的顺序为:14->15   2->3    5->7      4->5    9->14   18->19
  
  每次先处理结点的右子树,递归到叶子结点(没有左子树和右子树)。然后返回上一层,递归处理左子树。
  
  类似于前序遍历的镜像:  trave(root.right) ;
  
  trave(root.left);
  
  对每一层结点的处理可以通过tem=tem.next来进行跳转。
  
  得到如下方法:
  
  复制代码
  
  public class Solution www.gcyL157.com{
  
  public void connect(TreeLinkNode root) {
  
  if(root==null)return;
  
  TreeLinkNode tem=root;
  
  //处理root左子树的指针
  
  if(root.left!=www.mhylpt.com/  null){
  
  if(root.right!=null){
  
  root.left.next=root.right;
  
  }
  
  //处理root.left.next的指向
  
  while(tem.next!=null&&root.left.next==null){
  
  tem=tem.next;//tem可以在同一层结点进行跳转
  
  if(tem.left!=null)root.left.next=tem.left;
  
  else if(tem.right!=null)root.left.next=tem.right;
  
  }
  
  }
  
  //处理root右子树的指针
  
  if(root.right!=null){
  
  //处理root.right.next的指向
  
  while(tem.next!=null&&root.right.next==null){
  
  tem=tem.next;//tem可以在同一层结点进行跳转
  
  if(tem.left!=null)root.right.next=tem.left;
  
  else if(tem.right!=null)root.right.next=tem.right;
  
  }
  
  }
  
  connect(root.right);
  
  connect(root.left);
  
  }
  
  }
  
  复制代码
  
  结果:  执行用时: 1 ms, 在Populating Next Right Pointers in Each Node II的Java提交中击败了98.04% 的用户

猜你喜欢

转载自blog.csdn.net/li123128/article/details/84853636