[Python] SWEA D3 : 8016 - 홀수 피라미드

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=3&contestProbId=AWvzGUKKPVwDFASy&categoryId=AWvzGUKKPVwDFASy&categoryType=CODE&problemTitle=&orderBy=SUBMIT_COUNT&selectCodeLang=ALL&select-1=3&pageSize=10&pageIndex=4

SW Expert Academy

---

T = int(input())
for tc in range(1,1+T):
    n = int(input()) - 1
    first = n * (2*2 + (n-1)*4) // 2 + 1
    second = n * (2*6 + (n-1)*4) // 2 + 1
    print(f'#{tc} {first} {second}')

 

열심히 풀었는데 파이썬 제출이 불가능한 문제.. 아쉬워서 풀이 그냥 작성해본다.

시작점과 끝점이 두번째 줄 숫자에서 4씩 늘어난다는 점 이용해서 등차수열 이용해서 푼 문제 

- 시작점은 2, 끝점은 6으로 시작