添加 Swift 代码实现

zztian1024 · zztian1024 · commit 9116bc63ce06 · 2021-07-19T23:13:17.000+08:00
diff --git a/animation-simulation/二叉树/二叉树基础.md b/animation-simulation/二叉树/二叉树基础.md
@@ -406,6 +406,42 @@ public:
 };
 ```
 
+Swift Code：
+
+```swift
+class Solution {
+    func levelOrder(_ root: TreeNode?) -> [[Int]] {
+        var res:[[Int]] = []
+        guard root != nil else {
+            return res
+        }
+        var queue:[TreeNode?] = []
+        queue.append(root!)
+
+        while !queue.isEmpty {
+            let size = queue.count
+            var list:[Int] = []
+            
+            for i in 0..<size {
+                guard let node = queue.removeFirst() else {
+                    continue
+                }
+                if node.left != nil {
+                    queue.append(node.left)
+                }
+                if node.right != nil {
+                    queue.append(node.right);
+                }
+                list.append(node.val)
+            }
+            res.append(list)
+        }
+
+        return res
+    }
+}
+```
+
 时间复杂度：O（n） 空间复杂度：O（n）
 
 大家如果吃透了二叉树的层序遍历的话，可以顺手把下面几道题目解决掉，思路一致，甚至都不用拐弯
