Populating Next Right Pointers in Each Node II

一开始用递归,发现有些条件不满足,后来用队列的方法就accept了。

/**
 * Definition for binary tree with next pointer.
 * public class TreeLinkNode {
 *     int val;
 *     TreeLinkNode left, right, next;
 *     TreeLinkNode(int x) { val = x; }
 * }
 */
 
 import java.util.LinkedList;
    public class Solution {
        public void connect(TreeLinkNode root) {
        if(root==null){
    		 return;
    	 }
        LinkedList<TreeLinkNode> q=new LinkedList<TreeLinkNode>();
        TreeLinkNode p;
        q.offer(root);

       while((p=q.poll())!=null){
    	   work(p);
    	   if(p.left!=null){
    		   q.offer(p.left);
    	   }
    	   if(p.right!=null){
    		   q.offer(p.right);
    	   }
    	   
       }
    }
    
    
    public void work(TreeLinkNode root) {
        // Note: The Solution object is instantiated only once and is reused by each test case.
    	if(root==null){
     		return;
     	}
    	TreeLinkNode temp=root.next ;
    	while(temp!=null){
    		if(temp.left==null && temp.right==null){
    			temp=temp.next;
    		}
    		else{
    			temp=(temp.left==null)?temp.right:temp.left;
    			break;
    		}
    	}
    	
     	if(root.left!=null){
     		root.left.next=root.right==null?temp:root.right;
     	}
     	if(root.right!=null){
     		root.right.next=temp;
     	}
    }
    
    
    
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值