From 9b7e056fa5d90b614a1a4926a4e46577433fe3e1 Mon Sep 17 00:00:00 2001
From: marspere <1587393449@qq.com>
Date: Fri, 19 Aug 2022 13:56:47 +0800
Subject: [PATCH] =?UTF-8?q?Update=200102:=E4=BC=98=E5=8C=96=E4=BA=8C?=
 =?UTF-8?q?=E5=8F=89=E6=A0=91=E7=9A=84=E5=B1=82=E5=BA=8F=E9=81=8D=E5=8E=86?=
 =?UTF-8?q?=E4=B8=AD199=E9=A2=98=E7=9A=84go=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 problems/0102.二叉树的层序遍历.md | 35 +++++++++++------------
 1 file changed, 16 insertions(+), 19 deletions(-)

diff --git a/problems/0102.二叉树的层序遍历.md b/problems/0102.二叉树的层序遍历.md
index 25189dba..0063f36e 100644
--- a/problems/0102.二叉树的层序遍历.md
+++ b/problems/0102.二叉树的层序遍历.md
@@ -810,33 +810,30 @@ go:
 199. 二叉树的右视图
  */
 func rightSideView(root *TreeNode) []int {
-    queue:=list.New()
-    res:=[][]int{}
-    var finaRes []int
-    if root==nil{
-        return finaRes
+    if root == nil {
+        return nil
     }
+    res := make([]int, 0)
+    queue := list.New()
     queue.PushBack(root)
-    for queue.Len()>0{
-        length:=queue.Len()
-        tmp:=[]int{}
-        for i:=0;i<length;i++{
-            node:=queue.Remove(queue.Front()).(*TreeNode)
-            if node.Left!=nil{
+
+    for queue.Len() > 0 {
+        length := queue.Len()
+        for i := 0; i < length; i++ {
+            node := queue.Remove(queue.Front()).(*TreeNode)
+            if node.Left != nil {
                 queue.PushBack(node.Left)
             }
-            if node.Right!=nil{
+            if node.Right != nil {
                 queue.PushBack(node.Right)
             }
-            tmp=append(tmp,node.Val)
+            // 取每层的最后一个元素，添加到结果集中
+            if i == length-1 {
+                res = append(res, node.Val)
+            }
         }
-        res=append(res,tmp)
     }
-    //取每一层的最后一个元素
-    for i:=0;i<len(res);i++{
-        finaRes=append(finaRes,res[i][len(res[i])-1])
-    }
-    return finaRes
+    return res
 }
 ```