새소식

Computer Science/코딩테스트 문제 풀이

[예제] [파이썬] [탐색] 격자판 최대합

  • -

5*5 격자판에 아래롸 같이 숫자가 적혀있습니다.

 

N*N의 격자판이 주어지면 각 행의 합, 각 열의 합, 두 대각선의 합 중 가 장 큰 합을 출력합
니다.

 

▣ 입력설명
첫 줄에 자연수 N이 주어진다.(1<=N<=50)
두 번째 줄부터 N줄에 걸쳐 각 줄에 N개의 자연수가 주어진다. 각 자연수는 100을 넘지 않는
다.


▣ 출력설명
최대합을 출력합니다.


▣ 입력예제 1
5
10 13 10 12 15
12 39 30 23 11
11 25 50 53 15
19 27 29 37 27
19 13 30 13 19


▣ 출력예제 1
155

 


 

import sys
sys.stdin=open('input.txt','r')

def solution(board):
    rows_sum=[sum(arr) for arr in board]
    cols_sum=[sum(arr) for arr in zip(*board)]
    digs_sum=[sum(board[i][i] for i in range(N)),
              sum(board[i][N-i-1] for i in range(N))]
    
    return max(rows_sum+cols_sum+digs_sum)
    
if __name__=="__main__":
    N=int(input())
    board=[list(map(int, input().split())) for _ in range(N)]
    
    print(solution(board))
Contents

포스팅 주소를 복사했습니다

이 글이 도움이 되었다면 공감 부탁드립니다.