목차
격자판 최대합
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)
뭐 비슷하게 그렇게 깔끔해 보이진 않는다...!
탐색 시물레이션이다 보니 이렇게 더러운 코드도 꽤나 나온는듯 ㅇㅅㅇ
'코딩테스트[파이썬] > 알고리즘 문제풀이 입문' 카테고리의 다른 글
[탐색&시물레이션] - 곳감(모래시계) (2) | 2024.02.05 |
---|---|
[탐색&시물레이션] - 사과나무(다이아몬드) (2) | 2024.02.05 |
[탐색 & 시물레이션] - 수들의 합 (0) | 2024.02.05 |
[탐색&시물레이션] - 두 리스트 합치기 (2) | 2024.02.05 |
[탐색&시물레이션] - 숫자만 추출 (2) | 2024.02.05 |