Algorism/Python

격자판 최대합

aoaa 2022. 3. 14. 14:40

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

10 13 10 12 15
12 30 30 28 11
11 25 50 53 15
19 27 29 37 27
19 13 30 13 19

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

▣ 입력설명

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

▣ 출력설명

최대합을 출력합니다.

▣ 입력예제

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

n=int(input())
a=[list(map(int, input().split())) for _ in range(n)]
largest=0
for i in range(n):
    sum1=sum2=0
    for j in range(n):
        sum1+=a[i][j]
        sum2+=a[j][i]
    if sum1>largest:
        largest=sum1
    if sum2>largest:
        largest=sum2
sum1=sum2=0
for i in range(n):
    sum1+=a[i][i]
    sum2+=a[i][n-i-1]
    if sum1>largest:
        largest=sum1
    if sum2>largest:
        largest=sum2
print(largest)

a에 list문 안에서 변수없이(_) for문을 돌려줍니다. 그 후, a를 출력하면

[10, 13, 10, 12, 15] ... [19, 13, 30, 13, 19] 리스트 형태로 출렵됩니다.

가장 큰 값을 저장하기 위한 largest를 설정해줍니다. (0)

반복문을 실행하는데, n행과 n열 크기만큼 반복합니다. (문제같은 경우는 5)

sum1과 sum2에 행과 열의 개수만큼 반복문을 실행하여, 가장 큰 값을 largest에 저장해줍니다.

두 번째 반복문 에서는 대각선의 합중에서 가장 큰 값을 구하는 것입니다.

sum1은 i행, i열만큼 반복하게 되면 0번인덱스부터 4번인덱스까지 이어지는 대각선의 합을 구하게 되고,

sum2는 그와 반대인 대각선의 합을 구할 수 있습니다.

그 후, largest를 출력하면 정답입니다.

155

'Algorism > Python' 카테고리의 다른 글

이분탐색  (0) 2022.03.14
사과나무(다이아몬드)  (0) 2022.03.14
카드 역배치  (0) 2022.03.11
숫자열만 추출하기  (0) 2022.03.10
회문 문자열  (0) 2022.03.10