프로그래머스/Lv.2
[프로그래머스/Python] 삼각 달팽이
KimNang
2023. 3. 15. 01:49
문제 정보
제목 : 삼각 달팽이
난이도 : Lv.2
사용 언어 : Python
문제 링크
https://school.programmers.co.kr/learn/courses/30/lessons/68645
문제 설명
정수 n이 매개변수로 주어집니다. 다음 그림과 같이 밑변의 길이와 높이가 n인 삼각형에서 맨 위 꼭짓점부터 반시계 방향으로 달팽이 채우기를 진행한 후, 첫 행부터 마지막 행까지 모두 순서대로 합친 새로운 배열을 return 하도록 solution 함수를 완성해주세요.
제한 사항
- n은 1 이상 1,000 이하입니다.
입출력 예
n | result |
4 | [1,2,9,3,10,8,4,5,6,7] |
5 | [1,2,12,3,13,11,4,14,15,10,5,6,7,8,9] |
6 | [1,2,15,3,16,14,4,17,21,13,5,18,19,20,12,6,7,8,9,10,11] |
입출력 예 설명
입출력 예 #1
- 문제 예시와 같습니다.
입출력 예 #2
- 문제 예시와 같습니다.
입출력 예 #3
- 문제 예시와 같습니다.
나의 풀이
n번만큼 방향이 바뀌므로 n번 반복하는 for 루프를 생성합니다. 방향이 바뀔 때 마다 한 줄의 개수가 1씩 감소하고 있으므로 내부의 for 루프는 i부터 n까지 반복하도록 생성해줍니다. 바뀌는 방향이 아래,오른쪽, 위 3가지이므로 i를 3으로 나눈 나머지에 따라 좌표를 바뀌도록 했습니다. tmp[x][y]의 값에 num을 저장하고 num은 +1 해줍니다.
반복이 끝나면 answer에 tmp 리스트 중 값이 존재하면 그 값을 저장하고 리턴했습니다.
코드
def solution(n):
answer = []
x,y = -1,0
num = 1
tmp = [[0]*n for _ in range(n)]
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
tmp[x][y] = num
num += 1
for i in tmp :
for j in i :
if ( j ):
answer.append(j)
return answer