코딩테스트[파이썬]/알고리즘 문제풀이 입문

[탐색&시물레이션] - 격자판 최대합

softmoca__ 2024. 2. 5. 15:39
목차

 

격자판 최대합

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

 

나의 코드 

n=int(input())

arr=[list(map(int,input().split())) for _ in range(n) ]

S3=0 # 왼쪽아래 대각선
S2=0 # 오른 아래 대각선
S4=0 # 세로합
M=0

for i in range(n):
    if M<sum(arr[i]):
        M=sum(arr[i])
    S3+=arr[i][n-1-i]

for i in range(n):
    S4=0
    for j in range(n):
        if i==j:
            S2+=arr[i][j]
        S4=arr[j][i]
    if M<S4:
        M=S4

print(max(M,S3,S2))

 

음.. 이번 코드  별로다.. 일단 어떻게든 구하긴 했는데 분명더 깔끔함 코드가 있을꺼같은디..흠

 

 

정답 코드

import sys
sys.stdin = open("input.txt", 'r')
n=int(input())
a=[list(map(int, input().split())) for _ in range(n)]
largest=-2147000000
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)

뭐 비슷하게 그렇게 깔끔해 보이진 않는다...!

탐색 시물레이션이다 보니 이렇게 더러운 코드도 꽤나 나온는듯 ㅇㅅㅇ