Populating Next Right Pointers in Each Node 填充同一层的兄弟节点

给定一个二叉树

struct TreeLinkNode {
  TreeLinkNode *left;
  TreeLinkNode *right;
  TreeLinkNode *next;
}

填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL

初始状态下,所有 next 指针都被设置为 NULL

说明:

  • 你只能使用额外常数空间。
  • 使用递归解题也符合要求,本题中递归程序占用的栈空间不算做额外的空间复杂度。
  • 你可以假设它是一个完美二叉树(即所有叶子节点都在同一层,每个父节点都有两个子节点)。

示例:

给定完美二叉树,

     1
   /  \
  2    3
 / \  / \
4  5  6  7

调用你的函数后,该完美二叉树变为:

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

思路

这道题通过借助queue层序遍历,每次取栈顶元素然后next指针指向下一个栈顶元素即可(注意要控制每一层的尾部,否则尾部会指向下一层的头结点)。

    void connect(TreeLinkNode *root) {
	if (!root) {
		return;
	}
	queue<TreeLinkNode*> queue;
	queue.push(root);
	while (!queue.empty()) {
		int length = queue.size();
		for (int i = 0; i < length; i++) {
			TreeLinkNode *tmp = queue.front();
			queue.pop();
			if (i!=(length-1)) {
				tmp->next = queue.front();
			}
			if (tmp->left) {
				queue.push(tmp->left);
			}
			if (tmp->right) {
				queue.push(tmp->right);
			}
		}
	}        
    }




猜你喜欢

转载自blog.csdn.net/qq_26410101/article/details/80549546