728x90
문제 제목: 격자판 최대합
나의 풀이 (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부터 시작하므로
728x90
'coding test - python > 기본기 문제' 카테고리의 다른 글
문제 / 모래시계 / Python 파이썬 (0) | 2022.05.20 |
---|---|
문제 / 사과나무(다이아몬드) / Python 파이썬 (0) | 2022.05.19 |
문제 / *수들의 합/ Python (0) | 2022.04.05 |
문제 / 카드 역배치 / Python (0) | 2022.04.04 |
문제 / 두 리스트 합치기 / Python (0) | 2022.04.04 |