update 0116.填充每个节点的下一个右侧节点指针.md C++版本 一种更符合模板的层序遍历

Author: chengleqi

problems/0116.填充每个节点的下一个右侧节点指针.md

@@ -87,7 +87,7 @@ public:
 
 本题使用层序遍历是最为直观的，如果对层序遍历不了解，看这篇：[二叉树：层序遍历登场！](https://programmercarl.com/0102.二叉树的层序遍历.html)。
 
-层序遍历本来就是一层一层的去遍历，记录一层的头结点（nodePre），然后让nodePre指向当前遍历的节点就可以了。
+遍历每一行的时候，如果不是最后一个Node，则指向下一个Node；如果是最后一个Node，则指向nullptr。
 
 代码如下：
 
@@ -96,27 +96,18 @@ class Solution {
 public:
     Node* connect(Node* root) {
         queue<Node*> que;
-        if (root != NULL) que.push(root);
+        if (root != nullptr) que.push(root);
         while (!que.empty()) {
             int size = que.size();
-            vector<int> vec;
-            Node* nodePre;
-            Node* node;
-            for (int i = 0; i < size; i++) { // 开始每一层的遍历
-                if (i == 0) {
-                    nodePre = que.front(); // 记录一层的头结点
-                    que.pop();
-                    node = nodePre;
-                } else {
-                    node = que.front();
-                    que.pop();
-                    nodePre->next = node; // 本层前一个节点next指向本节点
-                    nodePre = nodePre->next;
-                }
-                if (node->left) que.push(node->left);
-                if (node->right) que.push(node->right);
+            for (int i = 0; i < size; ++i) {
+                Node* node = que.front();
+                que.pop();
+                if (i != size - 1) {
+                    node->next = que.front(); //如果不是最后一个Node 则指向下一个Node
+                } else node->next = nullptr;  //如果是最后一个Node 则指向nullptr
+                if (node->left != nullptr) que.push(node->left);
+                if (node->right != nullptr) que.push(node->right);
             }
-            nodePre->next = NULL; // 本层最后一个节点指向NULL
         }
         return root;
     }