문제 / 격자판 최대합 / Python

 

문제 제목: 격자판 최대합

 

 

---

나의 풀이 (80점)

n = int(input())
d = list(list(map(int, input().split())) for _ in range(n))
sum_list = [] # 각 행열,대각선의 합을 저장하는 리스트
cross1_list = [] #대각선을 저장하는 리스트
cross2_list = [] #반대 대각선을 저장하는 리스트

for i in range(n):
    sum_list.append(sum(d[i][:n])) #각 행들의 합
    sum_list.append(sum(d[:n][i])) #각 열들의 합
    cross1_list.append(d[i][i]) #대각선의 수
    cross2_list.append(d[i][n-i-1]) #반대대각선의 수
    
sum_list.append(sum(cross1_list)) #대각선의 합
sum_list.append(sum(cross2_list)) #반대 대각선의 합

print(max(sum_list)) #최댓값 출력

오토 채점 5번째에서 오류가나서 수동으로 값을 입력해보았다.

n의 값이 35일때는 잘 실행되었지만, n의 값이 50일때 값이 틀리게 나오는 것을 발견하였다..

 

 

모범답안

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] #행들의 합 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] #-1을 안하면 값에 오류가 남 리스트의 인덱스는 0부터 시작하므로
if sum1>largest: #최댓값 구하기
    largest=sum1
if sum2>largest:
    largest=sum2
print(largest) #최종 최댓값 구하기

---

※ 알아야 할 것

2차원 배열 값 입력받는 코드

- list(list(map(int, input().split())) for _ in range(n)) 

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

 

반대 대각선의 인덱스 주의하면서 범위 정할 것

ex) sum2+=a[i][n-i-1]

 #-1을 안하면 값에 오류가 남 리스트의 인덱스는 0부터 시작하므로