leetcode做题笔记116. 填充每个节点的下一个右侧节点指针

si_mple_2023-09-07 18:26

给定一个 完美二叉树，其所有叶子节点都在同一层，每个父节点都有两个子节点。二叉树定义如下：

复制代码

struct Node {
  int val;
  Node *left;
  Node *right;
  Node *next;
}

填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。

初始状态下，所有 next 指针都被设置为 NULL。

思路一：递归

cpp 复制代码

struct Node* connect(struct Node* root) {
	if(root==NULL)return NULL;
    else if(root->left == NULL || root->right == NULL) return root;

        root->left->next = root->right;
        struct Node* left = root->left->right;
        struct Node* right = root->right->left;
    while(left!=NULL&& right!=NULL){
        left->next = right;
        left = left->right;
        right = right->left;


    }
    root->left = connect(root->left);
        root->right = connect(root->right);
        return root;
    
}

分析：

本题要将每个next指针指向下一个右侧节点，可想到使用递归将左子树next指针指向right，再将右子树向下递归，最后输出root

总结：

本题考察递归的应用，将二叉树左子树指针指向右子树即可解决