这个题目其实算是比较简单的,第一个想法就是层次遍历。使用两个队列,一个存当前层的节点,一个存下一层的节点。这样交替的遍历使用,每个队列都会存一层的所有节点,然后出队列的同时改变next引用就行了。当然空间复杂度会高一点。但肯定还有更简单的方法,也就是递归处理。得益于大前提,这是一个完全二叉树,所以肯定所有的叶子节点都会在一层,这样就好做多了。
当然我们要明确一下更改next引用时有两种情况:
- 当前节点是父节点的左子节点。那此时只需要得到父节点的右子节点就好了。所以需要提供父节点这个参数从而获得右节点。
- 当前节点是父节点的右子节点,那此时需要父节点的相邻节点的左子节点,然后进行更改。
明确了这两种情况,那么问题就好解决多了,直接看代码清楚一点:
public Node connect(Node root) {
if(root == null)return null;root.next = null; //先把根节点处理一下helper(root, root.left);helper(root, root.right);return root;}private void helper(Node parent, Node root) {
if (root == null) //这时肯定是parent是个叶子节点,则直接返回return;if (parent.left == root) {
//当前节点是个左节点,则它的next指向相邻右节点root.next = parent.right; helper(root, root.left);helper(root, root.right);} else {
if (parent.next != null) {
//当前节点是个右节点,则它的next指向父节点相邻节点的左子节点root.next = parent.next.left;}helper(root, root.left);helper(root, root.right);}}
2019.12.3更新
每次重新做都有不同的发现,由于这个题目的特殊性(完全二叉树),所以根本不需要记住父节点,只需要记住一个根节点,然后直接对这个根节点的左右子树进行更改就行了,当遇到根节点是叶子节点的时候可以直接返回,因为在它的父节点已经完成了对叶子节点的next指针更新。下面的代码是更新后的,更加容易懂:
public Node connect(Node root) {
if (root == null)return root;helper(root);return root;}private void helper(Node root) {
//遇到叶子节点就可以直接返回,因为在它们的父节点已经完成了对叶子节点next指针的更新if (root.left == null && root.right == null) return;//先更改左子树next,直接把左子树的next指向右子树root.left.next = root.right;//再更改右子树next,需判断此根节点右边还有没有节点,有则是它右边节点(next)的左子树,没有则是nullif (root.next == null)root.right.next = null;else {
root.right.next = root.next.left;}//递归对左、右子树进行这样的更改helper(root.left);helper(root.right);}