반응형
출처 : 프로그래머스
코딩테스트 연습 - 삼각 달팽이
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]
programmers.co.kr
문제 설명
정수 n이 매개변수로 주어집니다. 다음 그림과 같이 밑변의 길이와 높이가 n인 삼각형에서 맨 위 꼭짓점부터 반시계 방향으로 달팽이 채우기를 진행한 후, 첫 행부터 마지막 행까지 모두 순서대로 합친 새로운 배열을 return 하도록 solution 함수를 완성해주세요.
제한사항
- n은 1 이상 1,000 이하입니다.
입출력 예nresult
| 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
- 문제 예시와 같습니다.
방향이 3개 있어서 3개 방향으로 생각 한다.
def solution(n):
answer = [[0 for i in range(n)] for j in range(n)]
cnt = 1
x, y = -1, 0
for i in range(n):
for j in range(i, n):
# 방향은 3개이다.
if i % 3 == 0:
x += 1
elif i % 3 == 1:
y += 1
else:
x -= 1
y -= 1
# 아래로 내려가기
answer[x][y] = cnt
cnt += 1
answer = [j for i in answer for j in i if j != 0]
return answer
n = 4
solution(n)
다른 사람 풀이 보고 sum(list, [])으로 수정하였다.
def solution(n):
answer = [[0]* i for i in range(1, n+1)]
cnt = 1
x, y = -1, 0
for i in range(n):
for j in range(i, n):
# 방향은 3개이다.
if i % 3 == 0:
x += 1
elif i % 3 == 1:
y += 1
else:
x -= 1
y -= 1
# 아래로 내려가기
answer[x][y] = cnt
cnt += 1
answer = sum(answer,[])
return answer반응형