LeetCode 116. 填充每个节点的下一个右侧节点指针

    技术2022-07-10  143

    116. 填充每个节点的下一个右侧节点指针

    解题思路:

    首先,这道题如果使用BFS很快就能AC,思路就是,使用队列存储每一层的节点,并用一个变量存储相应的层次,但是这样做的话,就违反了题目的 O ( 1 ) O(1) O(1) 的要求

    但是观察题目不难发现,这是个完美二叉树,既每个双亲节点都有两个孩子节点,而且如果某一层N已经链接完毕,则这一层的所有节点相当于一个链表,我们可以利用N层的节点来对N+1层进行处理,我们发现只有两类节点需要处理,既两个节点来自同一个父节点,以及两个节点来自不同父节点。 情况1(来自相同的父节点):这个很好办,node1和node2都来自同一个父节点parent,那就可以直接:parent->left->next = parent->right; 情况2(来自不同的父节点):这个需要利用parent的next才可以解决,也就是:parent->right->next = parent->next->left

    实现代码:
    /* // Definition for a Node. class Node { public: int val; Node* left; Node* right; Node* next; Node() : val(0), left(NULL), right(NULL), next(NULL) {} Node(int _val) : val(_val), left(NULL), right(NULL), next(NULL) {} Node(int _val, Node* _left, Node* _right, Node* _next) : val(_val), left(_left), right(_right), next(_next) {} }; */ class Solution { public: Node* connect(Node* root) { if(root == NULL){ return root; } Node *leftmost = root; while(leftmost->left != NULL){ Node *head = leftmost; while(head != NULL){ head->left->next = head->right; if(head->next != NULL){ head->right->next = head->next->left; } head = head->next; } leftmost = leftmost->left; } return root; } };
    Processed: 0.008, SQL: 9