[파이썬 코딩] Cos Pro 1급 1차 문제5 - 소용돌이 수

1 ] 문제

다음과 같이 n x n 크기의 격자에 1부터 n x n까지의 수가 하나씩 있습니다.
이때 수가 다음과 같은 순서로 배치되어있다면 이것을 n-소용돌이 수라고 부릅니다.
소용돌이 수에서 1행 1열부터 n 행 n 열까지 대각선상에 존재하는 수들의 합을 구해야 합니다.
위의 예에서 대각선상에 존재하는 수의 합은 15입니다.
격자의 크기 n이 주어질 때 n-소용돌이 수의 대각선상에 존재하는 수들의 합을 return 하도록 solution 함수를 완성해주세요.

 

 

 

2 ] 해답

# nXn크기의 격자모양에서 벗어나는 숫자가 없게 정의
def in_range(i, j, n):
   return 0 <= i and i < n and 0 <= j and j < n

def solution(n):
   pane = [[0 for j in range(n)] for i in range(n)]
   dy = [0, 1, 0, -1]
   dx = [1, 0, -1, 0]
   ci, cj = 0, 0
   num = 1
   while in_range(ci, cj, n) and pane[ci][cj] == 0:
       for k in range(4):
           if not in_range(ci, cj, n) or pane[ci][cj] != 0:
                break
           while True:
                 pane[ci][cj] = num
                 num += 1 
                 ni = ci + dy[k]
                 nj = cj + dx[k]
                 if not in_range(ni, nj, n) or pane[ni][nj] != 0:
                       #ci,cj 값 3x3격자이내의 인텍스값 정의 dy,dx
                       ci += dy[(k + 1) % 4]
                       cj += dx[(k + 1) % 4]
                       break
                ci = ni
                cj = nj

ans = 0
for i in range(n):
ans += pane[i][i]
return ans

ci, cj 값 / 격자모양 이내