[SWEA] 파이썬 2805. 농작물 수확하기

 

~목차~

문제

문제 해결 포인트

작성 코드

---

 

문제

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

SW Expert Academy

 

 

문제 해결 포인트

Q. 마름모를 어떻게 구분하여 찾아낼 것인가

 - 농장은 항상 홀수로 (1, 3, 5, 7, 9, ... ) NxN 형태를 나타낸다.

 - 그러므로 전체 길이의 중간 위치가 항상 존재하고 해당 위치(n//2)부터 마름모는 시작된다.

 - 마름모의 왼쪽 면을 시작위치(a)로 지정하고 해당 위치로 부터 얼만큼의 길이(b)만큼 수확할 것인지 행별로 생각해보자

 - 마름모 모양이 아닌 삼각형 두개의 합으로 생각해보면 n//2위치를 기준으로 완전히 동일한 형태를 띈다.

   해당 위치가 a와 b의 값이 달라지는 포인트이다.

 - a는 다음 행에서 한 칸 뒤로 가던 것이 한칸 앞으로 가게 되고, b는 2칸씩 넓혀가던 것을 2칸씩 좁혀가는 것이다.

 

 

작성 코드

 

for test in range(1, int(input())+1):
    n = int(input())  # nxn 농장
    board = []  # 농장
    result = 0  # 수확물
    for i in range(n):
        board.append(list(int(i) for i in input()))  # 문자열 입력값에 int씌워서 한 줄씩 리스트

    a, b = n//2, 1  # 각 행의 어느 위치(a)에서 얼만큼(b)
    for i in range(n):  # 행
        result += sum(board[i][a:a+b])
        if i < n//2:  # n의 절반 위치 까지는 위쪽 방향 삼각형
            a -= 1
            b += 2
        else:  # 절반 이하부터는 아래쪽 방향 삼각형
            a += 1
            b -= 2

    print("#{} {}".format(test, result))