您的位置：首页 > Web前端 > Node.js

5.1.10 Populating Next Right Pointers in Each Node II

2015-06-05 16:49 423 查看

Notes:
	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

	Solution: 1. iterative way with CONSTANT extra space.
	2. iterative way + queue. Contributed by SUN Mian(孙冕).
	3. recursive solution.
	*/

	/**
	* Definition for binary tree with next pointer.
	* struct TreeLinkNode {
	* int val;
	* TreeLinkNode left, right, *next;
	* TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}
	* };
	*/

void connect_2(TreeLinkNode *root) {
if (root == NULL) return;
queue<TreeLinkNode *> q;
q.push(root);
q.push(NULL);
TreeLinkNode *last = NULL;
TreeLinkNode dummy(-1);
TreeLinkNode *pre = &dummy;
while (!q.empty()) {
TreeLinkNode *node = q.front();
q.pop();
if (node == NULL) {
if (dummy.next) q.push(NULL);
pre = &dummy;
pre->next = NULL;
} else {
pre->next = node;
pre = pre->next;
if (node->left) q.push(node->left);
if (node->right) q.push(node->right);
}
}
}

内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理

标签：

相关文章推荐

新的分享

章节导航