[프로그래머스] 삼각 달팽이 (Python)
2023. 6. 8. 11:55ㆍ알고리즘/구현
https://school.programmers.co.kr/learn/courses/30/lessons/68645
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
def solution(n):
arr = [[0] * n for i in range(n)]
x,y=-1,0
a=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
arr[x][y] = a
a +=1
answer=[]
for i in arr:
for j in i:
if j != 0:
answer.append(j)
return answer
