[C++] LeetCode 116. 填充同一层的兄弟节点

题目

给定一个二叉树

填充它的每个 next指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。
初始状态下，所有 next指针都被设置为 NULL。
说明:

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

示例:
给定完美二叉树，

调用你的函数后，该完美二叉树变为：

题解

这道题首先考虑层序遍历，然后每一层向右连接。而二叉树的层序标准写法就是用queue+bfs。但是这道题要求常数空间，故queue就无法使用了，而题目说明递归不算空间。所以考虑用递归来实现，递归遍历二叉树左侧节点，每一层往右遍历当前层节点，同时让下一层子节点往右指。

代码

/**
 * 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==NULL||root->left==NULL) return;
        TreeLinkNode *node=root;
        while(root){
            root->left->next=root->right;
            if(root->next!=NULL){
                root->right->next=root->next->left;
            }
            root=root->next;
        }
        connect(node->left);
    }
};