정수를 나선형으로 배치하기

코딩테스트 연습 - 정수를 나선형으로 배치하기 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

 

나의 코드

def solution(n):

    answer = []

    for i in range(n):

        answer.append([0]*n)

    print(answer)

   

    # 4332211, 544332211 이런식으로 추가 되므로 리스트 만들어줌

    seq = [n]

    for i in range(n-1,0,-1):

        print(i)

        seq.append(i)

        seq.append(i)

    print(seq)

   

    # 초기 좌표, num

    x,y = 0,-1

    number = 0

   

    # answer값들을 조건에 따라 하나씩 채워준다,

    for i in range(len(seq)):

        for j in range(seq[i]):

            if i%4 == 0:

                y += 1

                number += 1

                answer[x][y] = number

            elif i%4 == 1:

                x += 1

                number += 1

                answer[x][y] = number

            elif i%4 == 2:

                y -= 1

                number += 1

                answer[x][y] = number

            elif i%4 == 3:

                x -= 1

                number += 1

                answer[x][y] = number

    print(answer)    

    return answer