코딩테스트/프로그래머스
[Programmers] 삼각 달팽이 - 월간 코드 챌린지1 (python)
NUAR
2020. 9. 16. 23:45
삼각 달팽이 - 프로그래머스
def solution(n):
li = [[0] * n for _ in range(n)]
ret = []
x, y = -1, 0
num = 1
for i in range(n):
for j in range(i, n):
if i % 3 == 0:
x += 1
elif i % 3 == 1:
y += 1
elif i % 3 == 2:
x -= 1
y -= 1
li[x][y] = num
num += 1
for i in li:
for j in i:
if j != 0:
ret.append(j)
return ret