문제
N X N크기의 농장이 있다.
이 농장에는 이상한 규칙이 있다.
규칙은 다음과 같다.
① 농장은 크기는 항상 홀수이다. (1 X 1, 3 X 3 … 49 X 49)
② 수확은 항상 농장의 크기에 딱 맞는 정사각형 마름모 형태로만 가능하다.
1 X 1크기의 농장에서 자라는 농작물을 수확하여 얻을 수 있는 수익은 3이다.
3 X 3크기의 농장에서 자라는 농작물을 수확하여 얻을 수 있는 수익은 16 (3 + 2 + 5 + 4 + 2)이다.
5 X 5크기의 농장에서 자라는 농작물의 수확하여 얻을 수 있는 수익은 25 (3 + 2 + 1 + 1 + 2 + 5 + 1 + 1 + 3 + 3 + 2 + 1)이다.
농장의 크기 N와 농작물의 가치가 주어질 때, 규칙에 따라 얻을 수 있는 수익은 얼마인지 구하여라.
[제약 사항]
농장의 크기 N은 1 이상 49 이하의 홀수이다. (1 ≤ N ≤ 49)
농작물의 가치는 0~5이다.
[입력]
가장 첫 줄에는 테스트 케이스의 개수 T가 주어지고, 그 아래로 각 테스트 케이스가 주어진다.
각 테스트 케이스에는 농장의 크기 N과 농장 내 농작물의 가치가 주어진다.
[출력]
각 줄은 '#t'로 시작하고, 공백으로 농장의 규칙에 따라 얻을 수 있는 수익을 출력한다.
(t는 테스트 케이스의 번호를 의미하며 1부터 시작한다.)
풀이
농장의 크기 N과 farm이라는 2차원 배열을 입력으로 받는다.
a를 중간지점 인덱스라고 둔다 a = N//2
그리고 sum1에 첫번째 행에서는 중간값, 두번째 행에서는 중간값-1,중간값+1, 세번째 행에서는 중간값-2, 중간값+2 범위까지의 숫자들을 다 더해준다.
#1 중간까지의 반복문과 #2 중간이후의 반복문을 나눠주었다.
sum1을 출력한다.
코드
T = int(input())
for test_case in range(1,T+1):
N = int(input())
farm = [list(map(int, input())) for _ in range(N)]
sum1=0
a = N//2
k=0
for i in range(0,a+1): #1 중간까지
sum1 += sum(farm[k][a-i:a+i+1])
k+=1
for i in range(1,a+1): #2 중간이후
sum1 += sum(farm[k][0+i:N-i])
k+=1
print(f'#{test_case} {sum1}')