小编典典

在螺旋中循环

all

一位朋友需要一种算法,可以让他遍历 NxM 矩阵的元素(N 和 M 是奇数)。我想出了一个解决方案,但我想看看我的同胞们是否能想出一个更好的解决方案。

我发布我的解决方案作为这个问题的答案。

示例输出:

对于 3x3 矩阵,输出应为:

(0, 0) (1, 0) (1, 1) (0, 1) (-1, 1) (-1, 0) (-1, -1) (0, -1) (1, -1 )

3x3 矩阵

此外,该算法应支持非方阵,例如对于 5x3 矩阵,输出应为:

(0, 0) (1, 0) (1, 1) (0, 1) (-1, 1) (-1, 0) (-1, -1) (0, -1) (1, -1 ) (2, -1)
(2, 0) (2, 1) (-2, 1) (-2, 0) (-2, -1)

5x3 矩阵


阅读 61

收藏
2022-08-27

共1个答案

小编典典

这是我的解决方案(在 Python 中):

def spiral(X, Y):
    x = y = 0
    dx = 0
    dy = -1
    for i in range(max(X, Y)**2):
        if (-X/2 < x <= X/2) and (-Y/2 < y <= Y/2):
            print (x, y)
            # DO STUFF...
        if x == y or (x < 0 and x == -y) or (x > 0 and x == 1-y):
            dx, dy = -dy, dx
        x, y = x+dx, y+dy
2022-08-27