[SWEA][Python]#2805. 농작물 수확하기

📌풀이


내가 쓴 풀이(성공)

T = int(input())

for test_case in range(1, T+1):
    size = int(input())
    mid = size// 2
    result = 0
    
    for i in range(size):
        now = input()
        farm_line = list(map(int,list(now)))
        if i <= mid :
            result += sum(farm_line[mid-i:mid+i+1])
        else:
            result += sum(farm_line[mid-(size-i-1):mid+(size-i-1)+1])
    print(f'#{test_case} {result}')
  • 농작물의 합을 구할 때, 규칙성이 있다는 점을 활용
  • 인덱스의 중간값을 기준으로, sum을 하는 범위를 다르게 적용


📌후기


농작물의 합을 더하는 과정에서 규칙성을 활용해 조금 더 괜찮은 코드를 작성한 느낌! 처음에 농작물들을 모아놓은 리스트를 생성하려고 했는데, 굳이 필요하지 않음을 인지하고 수정했다!

좋은 웹페이지 즐겨찾기