leetcode：Populating Next Right Pointers in Each Node（常数空间，连接二叉树每一层所有节点）【面试算法题】

最新推荐文章于 2020-07-21 16:47:39 发布

原创最新推荐文章于 2020-07-21 16:47:39 发布 · 3.9k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #二叉树 #dfs

leetcode题解面试算法同时被 2 个专栏收录

76 篇文章

订阅专栏

leetcode面试算法题

75 篇文章

订阅专栏

本文介绍了一种解决完全二叉树中各层节点连接的问题，通过递归方式将每层的所有节点指针指向其右侧节点，实现仅使用常数额外空间完成节点间的连接。

题目：

Given a binary tree

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

Populate each next pointer to point to its next right node. If there is no next right node, the next pointer should be set to NULL.

Initially, all next pointers are set to NULL.

Note:

You may only use constant extra space.
You may assume that it is a perfect binary tree (ie, all leaves are at the same level, and every parent has two children).

For example,
Given the following perfect binary tree,

After calling your function, the tree should look like:

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

题意把给定的一个完全二叉树每一层所有的节点连接起来，只能用常数的额外空间。

用递归的做法，把当前root的左节点和它的所有深度下的右节点与右节点和它的所有深度下的左节点连接。

目的就是把因为当前root左右节点而分开的子节点全部连接起来。

画个图就很容易发现，这样做就把需要连接的节点在他们的公共祖先root的递归层中一起处理。

/**
 * 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) {}
 * };
 */
class Solution {
public:
    void connect(TreeLinkNode *root) {
        if(!root)return;
        if(root->left)
        {
            TreeLinkNode *l=root->left,*r=root->right;
            while(l!=NULL)
            {
                l->next=r;
                l=l->right;
                r=r->left;
            }
            connect(root->left);
            connect(root->right);
        }
    }
};
// blog.csdn.net/havenoidea