欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-05-20 16:17:59
...

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

解题思路:

首先,这道题如果使用BFS很快就能AC,思路就是,使用队列存储每一层的节点,并用一个变量存储相应的层次,但是这样做的话,就违反了题目的 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;
    }
};
相关标签: leetcode随笔