[Python] 백준 2566 최댓값 | NameError

<그림 1>과 같이 9×9 격자판에 쓰여진 81개의 자연수 또는 0이 주어질 때, 이들 중 최댓값을 찾고 그 최댓값이 몇 행 몇 열에 위치한 수인지 구하는 프로그램을 작성하시오.

예를 들어, 다음과 같이 81개의 수가 주어지면

이들 중 최댓값은 90이고, 이 값은 5행 7열에 위치한다.

 

 

 

 

처음 제출한 코드를 먼저 보겠습니다.

*틀린 코드 입니다

maxN = 0
row = 0
col = 0

for i in range(9):
    nlist = list(map(int, input().split()))
    for j in range(9):
        if nlist[j] > maxN:
            maxN = nlist[j]
            row = i+1
            col = j+1

print(maxN)
print(row, end=" ")
print(col)

무엇이 잘못 되었을까요?

저는 무의미한 for문의 중첩이 문제라고 생각했었는데 그 문제 때문이었을까요?

문제로 다시 돌아가 보겠습니다.

질문에서 각 셀의 값은 0또는 자연수의 값을 갖는다고 했습니다.

 

모든 셀의 값이 0을 갖는다면 row와 col은 어떤 위치 값을 가져야 할까요?

0, 0은 아닐겁니다. 

 

그러므로 row와 col의 위치에는 첫번째 셀의 위치값(1, 1)을 입력해주고 시작해야합니다

 

 

다음은 수정된 코드입니다.

앞서 말했듯 무의미한 for문을 제거하고,

한 행을 읽었을 때 최대값(max 함수)과의 비교를 통해 시간을 단축시켰습니다.

maxN = 0
row = 1
col = 1

for i in range(9):
    nlist = list(map(int, input().split()))
    if max(nlist) > maxN:
        maxN = max(nlist)
        row = i + 1
        col = nlist.index(maxN) + 1

print(maxN)
print(row, col)