[swea] 2805 농작물 수확하기(파이썬/python)

🎈문제

 

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

🎁알고리즘 및 접근

단순 구현

1. 행렬에 index로 접근하는 단순한 문제이다.

2. 규칙을 찾아야 된다.

3. 위쪽 / 가운데 / 아래쪽으로 나눠서 규칙을 찾았다.

 

💻코드

# 2805 농작물 수확하기


T = int(input())

for test_case in range(1,T+1):
    n = int(input())
    farm = [list(map(int,list(input()))) for _ in range(n)]

    invest = 0
    # 위쪽
    for i in range(n//2):
        invest += sum(farm[i][n//2-i:n//2+i+1]) # [0,3:4] , [1,2:5] , [2,1:6]
    # 가운데
    invest += sum(farm[n//2])
    # 아래쪽
    for i in range(n-1,n//2,-1):    # [6,3] -> 6,5,4
        invest += sum(farm[i][n//2-(n-1-i):n//2+(n-i)])      # [6,3:4] , [5,2:3+(6-5)]
    
    print(f"#{test_case} {invest}")

🎄코드 설명

주석으로 설명은 대체한다.