[Python] 삼각달팽이

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]

 

---

def solution(n):
    answer = []
    tri = [[0]*i for i in range(1, n+1)]
    i=0
    j=0
    cnt=1
    max = n*(n+1)//2
    tri[i][j] = cnt
    while cnt < max :
        while i+1 < n and tri[i+1][j] == 0:
            i += 1
            cnt += 1
            tri[i][j] = cnt
        while j+1 < n and tri[i][j+1] == 0:
            j += 1
            cnt += 1
            tri[i][j] = cnt
        while i-1 < n and j-1<n and tri[i-1][j-1] == 0:
            i -= 1
            j -= 1
            cnt += 1
            tri[i][j] = cnt
    for a in tri:
        answer.extend(a)
    return answer