码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode 117:Populating Next Right Pointers in Each Node II

时间:2017-06-09 13:25:14      阅读:182      评论:0      收藏:0      [点我收藏+]

标签:data   connect   for   while   call   content   nts   span   data-   

Follow up for problem "Populating Next Right Pointers in Each Node".

What if the given tree could be any binary tree? Would your previous solution still work?

Note:

  • You may only use constant extra space.

For example,
Given the following binary tree,

         1
       /        2    3
     / \        4   5    7

After calling your function, the tree should look like:

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

Subscribe to see which companies asked this question

//二叉树进行层次遍历
//推断最底层的条件:假设当前訪问的层有任一节点存在子节点,说明当前层不是最底层。
class Solution{
public:
	TreeLinkNode* nextLeft;
	bool hasToTheEnd;
	void connect(TreeLinkNode *root)
	{
		if (!root)  return;

		hasToTheEnd = false;
		int level = 1;
		while (!hasToTheEnd)
		{
			hasToTheEnd= true;
			nextLeft= NULL;
			VisitLevel(root, level);
			++level;
		}
	}

	void VisitLevel(TreeLinkNode* node, int level)
	{
		if (level == 1)
		{
			if (nextLeft != NULL)
			{
				nextLeft->next = node;
			}
			nextLeft = node;

			if (node->left != NULL || node->right != NULL)
			{
				hasToTheEnd= false;
			}
		}

		if (node->left)	  VisitLevel(node->left, level - 1);
		if (node->right)  VisitLevel(node->right, level - 1);
	}
};

技术分享

LeetCode 117:Populating Next Right Pointers in Each Node II

标签:data   connect   for   while   call   content   nts   span   data-   

原文地址:http://www.cnblogs.com/mthoutai/p/6971416.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!