文章目錄

• • 1. 題目
  • 2. 解題
  • • 2.1 遞歸
    • 2.2 循環
    • 2.3 O(1)空間復雜度

1. 題目

給定一個完美二叉樹，其所有葉子節點都在同一層，每個父節點都有兩個子節點。二叉樹定義如下：

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

填充它的每個 next 指針，讓這個指針指向其下一個右側節點。如果找不到下一個右側節點，則將 next 指針設置為 NULL。

初始狀態下，所有 next 指針都被設置為 NULL

2. 解題

2.1 遞歸

class Solution { public:Node* connect(Node* root) {if(root == NULL || root->left == NULL)return root;root->left->next = root->right;if(root->next != NULL)root->right->next = root->next->left;connect(root->left);connect(root->right);return root;} };

2.2 循環

class Solution { public:Node* connect(Node* root) {if(root == NULL || root->left == NULL)return root;queue<Node*> q;q.push(root);Node *p;int n;while(!q.empty()){n = q.size();while(n--){p = q.front();if(p->left != NULL){q.push(p->left);q.push(p->right);p->left->next = p->right;if(p->next != NULL)p->right->next = p->next->left;}q.pop();} }return root;} };

2.3 O(1)空間復雜度

• 利用next指針

class Solution { public:Node* connect(Node* root) {if(root == NULL || root->left == NULL)return root;Node *parent = root, *lvfirst, *prev;while(parent != NULL){lvfirst = parent->left;if(lvfirst != NULL){while(parent){parent->left->next = parent->right;prev = parent->right;parent = parent->next;if(parent)prev->next = parent->left;}}parent = lvfirst; }return root;} };

創作挑戰賽新人創作獎勵來咯，堅持創作打卡瓜分現金大獎

總結

以上是生活随笔為你收集整理的LeetCode 116. 填充每个节点的下一个右侧节点指针（递归循环）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。