LeetCode - #59 螺旋矩阵 II

前言

我们社区陆续会将顾毅（Netflix 增长黑客，《iOS 面试之道》作者，ACE 职业健身教练。）的 Swift 算法题题解整理为文字版以方便大家学习与阅读。

LeetCode 算法到目前我们已经更新了 58 期，我们会保持更新时间和进度（周一、周三、周五早上 9:00 发布），每期的内容不多，我们希望大家可以在上班路上阅读，长久积累会有很大提升。

不积跬步，无以至千里；不积小流，无以成江海，Swift社区 伴你前行。如果大家有建议和意见欢迎在文末留言，我们会尽力满足大家的需求。

难度水平：中等

1. 描述

给你一个正整数 n ，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。

2. 示例

示例 1

输入：n = 3
输出：[[1,2,3],[8,9,4],[7,6,5]]

示例 2

输入：n = 1
输出：[[1]]

约束条件：

• 1 <= n <= 20

3. 答案

class SpiralMatrixII {
    func generateMatrix(_ n: Int) -> [[Int]] {
        guard n > 0 else {
            return [[Int]]()
        }
    
        var num = 1
        var res = Array(repeating: Array(repeating: 0, count: n), count: n)
        
        for layer in 0..<n / 2 {
            let start = layer
            let end = n - layer - 1
            
            // top
            for i in start..<end {
                res[start][i] = num
                num += 1
            }
            
            // right
            for i in start..<end {
                res[i][end] = num
                num += 1
            }
            
            // bottom
            for i in stride(from: end, to: start, by: -1) {
                res[end][i] = num
                num += 1
            }
            
            // left
            for i in stride(from: end, to: start, by: -1) {
                res[i][start] = num
                num += 1
            }
        }
        
        // handle the center one
        if n % 2 != 0 {
            res[n / 2][n / 2] = n * n
        }
        
        return res
    }
}

• 主要思想：从顺时针开始，填充数字，记住处理中间的一个。
• 时间复杂度：O(n^2)
• 空间复杂度：O(1)

该算法题解的仓库：LeetCode-Swift[1]

点击前往 LeetCode[2] 练习

关于我们

Swift社区是由 Swift 爱好者共同维护的公益组织，我们在国内以微信公众号的运营为主，我们会分享以 Swift实战、SwiftUl、Swift基础为核心的技术内容，也整理收集优秀的学习资料。

特别感谢 Swift社区 编辑部的每一位编辑，感谢大家的辛苦付出，为 Swift社区 提供优质内容，为 Swift 语言的发展贡献自己的力量，排名不分先后： 张安宇@微软[3]、戴铭@快手[4]、展菲@ESP[5]、倪瑶@Trip.com[6]、杜鑫瑶@新浪[7]、韦弦@Gwell[8]、张浩@讯飞[9]、张星宇@ByteDance[10]、郭英东@便利蜂[11]、何敏[12]

参考资料

[1]

LeetCode-Swift: https://github.com/soapyigu/LeetCode-Swift

[2]

LeetCode: https://leetcode.com/problems/spiral-matrix-ii/

[3]

张安宇: https://blog.csdn.net/mobanchengshuang

[4]

戴铭: https://ming1016.github.io

[5]

展菲: https://github.com/fanbaoying

[6]

倪瑶: https://github.com/niyaoyao

[7]

杜鑫瑶: https://weibo.com/u/3878455011

[8]

韦弦: https://www.jianshu.com/u/855d6ea2b3d1

[9]

张浩: https://github.com/zhanghao19920218

[10]

张星宇: https://github.com/bestswifter

[11]

郭英东: https://github.com/EmingK

[12]

何敏: https://weibo.com/3483803314/profile?rightmod=1&wvr=6&mod=personinfo&is_all=1

- EOF -