Leetcode行列IIコイル[]アレイ（59）

タイトル

正の整数nを考えると、1 n2はすべての要素を含ん生成し、要素が螺旋状に時計回りに次の正方行列に配置します。

例：

输入: 3
输出:
[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

答え

検索と同様に、表現についての上下移動の配列の定義。コードの量は、遼寧省を泣くことができます...

z = [
            [0, 1],  # 右
            [1, 0],  # 下
            [0, -1],  # 左
            [-1, 0]  # 上
        ]

コードの実装：

class Solution:
    # Time: O(N)，N为数值个数
    def generateMatrix(self, n):
        if n == 0:
            return []
        ans = [[-1]*n for _ in range(n)]  # 初始化为-1
        z = [
            [0, 1],  # 右
            [1, 0],  # 下
            [0, -1],  # 左
            [-1, 0]  # 上
        ]
        x = y = 0
        cur = 1
        while -1 in ans[n//2]:
            if ans[x][y] == -1:
                ans[x][y] = cur
                cur += 1
            while x + z[0][0] < n and y + z[0][1] < n and ans[x+z[0][0]][y+z[0][1]] == -1:  # 向右
                x = x + z[0][0]
                y = y + z[0][1]
                ans[x][y] = cur
                cur += 1
            while x + z[1][0] < n and y + z[1][1] < n and ans[x+z[1][0]][y+z[1][1]] == -1:  # 向下
                x = x + z[1][0]
                y = y + z[1][1]
                ans[x][y] = cur
                cur += 1
            while x + z[2][0] < n and y + z[2][1] < n and y + z[2][1] >= 0 and ans[x+z[2][0]][y+z[2][1]] == -1:  # 向左
                x = x + z[2][0]
                y = y + z[2][1]
                ans[x][y] = cur
                cur += 1
            while x + z[3][0] < n and x + z[3][0] >= 0 and y + z[3][1] < n and ans[x+z[3][0]][y+z[3][1]] == -1:  # 向上
                x = x + z[3][0]
                y = y + z[3][1]
                ans[x][y] = cur
                cur += 1
        return ans


s = Solution()
ans = s.generateMatrix(3)
print(ans)


# [
#  [ 1, 2, 3 ],
#  [ 8, 9, 4 ],
#  [ 7, 6, 5 ]
# ]