（java）美团一面之（Populating Next Right Pointers in Each Node）

最新推荐文章于 2024-09-21 10:38:43 发布

只对西红柿鸡蛋有感觉

最新推荐文章于 2024-09-21 10:38:43 发布

阅读量278

点赞数

文章标签： java 编程二叉树

本文链接：https://blog.csdn.net/chaiqunxing51/article/details/51579213

版权

题意就是给一棵二叉树，让你变成这个样子：

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

思路是：层次遍历+添加next节点

public class Solution {
    public void connect(TreeLinkNode root) {
        Queue<TreeLinkNode> q=new LinkedList<TreeLinkNode>();
	        int count=0;
	        int temp=0;
	        if(root!=null)
	        {
	        	q.offer(root);
	        	count++;
	        }
	        while(!q.isEmpty()){
	        	TreeLinkNode tempnode=q.poll();
	        	count--;
	        	tempnode.next=null;
	        	if(tempnode.left!=null){
        			temp++;
        			q.offer(tempnode.left);
        		}
        		if(tempnode.right!=null){
        			temp++;
        			q.offer(tempnode.right);
        		}
	        	while(count>0){
	        		TreeLinkNode tempnode2=q.poll();
	        		tempnode.next=tempnode2;
	        		tempnode=tempnode.next;
	        		if(tempnode.left!=null){
	        			temp++;
	        			q.offer(tempnode.left);
	        		}
	        		if(tempnode.right!=null){
	        			temp++;
	        			q.offer(tempnode.right);
	        		}
	        		count--;
	        	}
	        	count=temp;
	        	temp=0;
	        }
    }
}