[Cos Pro 1급] 1차 5번 - 소용돌이 수의 대각선의 합

문제5

다음과 같이 n x n 크기의 격자에 1부터 n x n까지의 수가 하나씩 있습니다.

 

 

이때 수가 가운데 사진과 같은 순서로 배치되어있다면 이것을 n-소용돌이 수라고 부릅니다.

소용돌이 수에서 1행 1열부터 n 행 n 열까지 대각선상에 존재하는 수들의 합을 구해야 합니다.

위의 예에서 대각선상에 존재하는 수의 합은 15입니다. 격자의 크기 n이 주어질 때

 n-소용돌이 수의 대각선상에 존재하는  수들의 합을 return 하도록 solution 함수를 완성해주세요.

 

매개변수 설명

격자의 크기 n이 solution 함수의 매개변수로 주어집니다.  n은 1 이상 100 이하의 자연수입니다.

 

return 값 설명

n-소용돌이 수의 대각선상에 존재하는 수들의 합을

return 해주세요.

 

예시

n	return
3	15
2	4

 

 

Python

def solution(n):
    total=1*n

    for i in range(n, 1,-2):
        total += (i-1)*2 * ((i-1)*2-1)

    return total

for i in range(2,8):
    print(i, solution(i))