LeetCode116 Populating Next Right Pointers in Each Node

题目链接：

https://leetcode.com/problems/populating-next-right-pointers-in-each-node/

题目描述：

给一个满二叉树。

树的结构。

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

每一个节点的next初始都为空

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

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

题目分析：

题目要求的是空间复杂度为O(1)。所以利用队列进行广搜是不行的。

我们可以利用树结构中的next。用一个外循环控制层，外循环实现每层的开始，然而用next可以遍历每一层的所有节点，在遍历那一层节点时，要把下一层的节点next链好。

主要思路是，假设root指的是2。

root->left->next=root->right;
root->right->next=root->next->left;

代码：

class Solution {
public:
    void connect(TreeLinkNode *root) {
       while(root!=NULL && root->left!=NULL){
           TreeLinkNode* ptr=root;
           while(ptr!=NULL){
               ptr->left->next=ptr->right;
               if(ptr->next!=NULL){
                   ptr->right->next=ptr->next->left;
               }
               ptr=ptr->next;
           }
           root=root->left;
       }
    }
};