leetcode Populating Next Right Pointers in Each Node II

이 문제는 이전 문제Populating Next Right Pointers in Each Node와 다르다. 여기서 요구하는 것은 일반적인 나무이기 때문에 난이도가 비교적 크다.이전에는 간단하게 왼쪽의 마지막과 오른쪽의 맨 왼쪽 링크를 찾으면 됩니다.현재 존재하는 문제는 오른쪽의 좌우 1층이 가장 왼쪽이 아니거나 길이가 같지 않다는 것이다.

        1

       /  \

      2    3

     / \    \

    4   5    7
 ：

         1 -> NULL

       /  \

      2 -> 3 -> NULL

     / \    \

    4-> 5 -> 7 -> NULL

그림: 만약에 2를 루트로 처리해야 할 때 우리는 2의 오른쪽 나무의 존재 여부를 판단해야 한다. 만약에 존재한다면 우리는 2의 넥스트 나무에서 5의 넥스트를 찾아야 한다. 그것은 틀림없이 3의 왼쪽 나무가 우선이지만 3의 오른쪽 나무가 없기 때문에 3의 오른쪽 나무가 7을 5의 넥스트로 판단하고 3의 오른쪽 나무도 비어 있다면 5의 넥스트가 없는 것이 아닐까, 아니,우리는 3의next가 비어 있는지 계속 고려해야 한다. 만약에 3의next가 비어 있지 않다면 3의next와 유사하게 좌우 트리를 판단하여 5에 부합되는next가 있는지 확인해야 한다.만약 아직 없다면 넥스트의 넥스트는 찾을 때까지, 혹은 넥스트가 비어 있을 때까지.
5의next처리 과정에서 먼저 3이 처리되었음을 요구해야 하기 때문에 먼저 오른쪽 트리로 돌아가야 한다.
5의 넥스트 처리 후 2의 왼쪽 트리가 비어 있는지 판단하고, 비어 있으면 되돌아오고, 비어 있지 않으면 5를 4의 넥스트에 주고, 5가 없으면 아까 찾은 5의 넥스트를 4로 삼는다.
그러면 다음과 같은 것들이 있습니다.

/**

 * 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 ;

        TreeLinkNode *p = root -> next;

        TreeLinkNode *sonNext = NULL;

        while (p)// next ， root next

        {

            if (p -> left)

            {

                sonNext = p -> left;

                break;

            }

            else if (p -> right)

            {

                sonNext = p -> right;

                break;

            }

            else

                p = p -> next;

        }

        if (root -> right)

        {

            root -> right -> next = sonNext;

            if (root -> left)

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

        }

        else if (root -> left)

            root -> left -> next = sonNext;

        

        connect(root -> right);

        connect(root -> left);

    }

};

비귀속 방법도 사용할 수 있다. 왜냐하면 이전의 모든 층이next를 가로로 연결할 수 있기 때문에 체인 테이블로 간주하거나 대열로 간주할 수 있다.그리고 매번 이 층의 기점부터 판단한다.단지 고려해야 할 요소가 많을 뿐이다. 다음 층의 시작점을 어떻게 기록하는지, 그리고 현재의next가 비어 있는지, 그녀의 좌우 나무가 비어 있는지, 그렇다면next의next는?

/**

 * 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 ;

        TreeLinkNode *son = NULL, *nextLayerBegin = NULL;

        bool flag = true; //  ， 

        while(root)

        {

            while(root)

            {

                if (!root -> left && !root-> right){ root = root -> next;continue;}

                if (root -> left)

                    son = root -> left;

                else

                    son = root -> right;

                if (flag)

                {

                    nextLayerBegin = son; //  

                    flag = false;

                }

                if (root -> left && root -> right)

                {son -> next = root -> right; son = root -> right;}

                if (root -> next)

                {

                    while (root -> next)

                    {

                        if (root -> next -> left)

                        {son -> next = root -> next -> left; son = root -> next -> left;root = root->next;break;}

                        else if (root -> next -> right)

                        {son -> next = root -> next -> right; son = root -> next -> right;root = root->next;break;}

                        else

                            root = root -> next;

                    }

                }

                else break;

            }

            root = nextLayerBegin; //  

            nextLayerBegin = NULL; // clear， 

            flag = true;

        }

    }

};