给定一个二叉树
struct Node { int val; Node *left; Node *right; Node *next; }填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。
初始状态下,所有 next 指针都被设置为 NULL。
示例:
输入:{"$id":"1","left":{"$id":"2","left":{"$id":"3","left":null,"next":null,"right":null,"val":4},"next":null,"right":{"$id":"4","left":null,"next":null,"right":null,"val":5},"val":2},"next":null,"right":{"$id":"5","left":null,"next":null,"right":{"$id":"6","left":null,"next":null,"right":null,"val":7},"val":3},"val":1} 输出:{"$id":"1","left":{"$id":"2","left":{"$id":"3","left":null,"next":{"$id":"4","left":null,"next":{"$id":"5","left":null,"next":null,"right":null,"val":7},"right":null,"val":5},"right":null,"val":4},"next":{"$id":"6","left":null,"next":null,"right":{"$ref":"5"},"val":3},"right":{"$ref":"4"},"val":2},"next":null,"right":{"$ref":"6"},"val":1} 解释:给定二叉树如图 A 所示,你的函数应该填充它的每个 next 指针,以指向其下一个右侧节点,如图 B 所示。
提示:
你只能使用常量级额外空间。使用递归解题也符合要求,本题中递归程序占用的栈空间不算做额外的空间复杂度。第一种思路:
层序遍历,跟LeetCode-Python-116. 填充每个节点的下一个右侧节点指针的第一种思路一样。
第二种思路:
递归,分类讨论:
当一个Node既有左孩子又有右孩子的时候,显然左孩子的next就是右孩子,右孩子的next得在Node的next里往右找,直到某个节点有孩子,这个孩子就是Node.right的next。
当一个Node只有左孩子的时候,左孩子的next得在Node的next里往右找。
当一个Node只有右孩子的时候,右孩子的next得在Node的next里往右找。
所以不妨用一个辅助函数findCousin(node, parent)来找到node.next,其中node是parent的子节点
对于一个需要找Cousin节点的node, 首先看叔叔节点parent.next有没有孩子,
如果有左孩子,那么很好这个左孩子就是结果,
如果没有左孩子但是有右孩子,那么也不错,这个右孩子是结果,
如果parent.next没有孩子,那么糟了,还得继续找更小的叔叔,即parent.next.next……
一直这么找下去,直到找到一个有孩子的叔叔,那么返回这个孩子,或者找完所有的叔叔,它们都没有孩子……
