[백준/Python] 2566 - 최댓값

Frye 'de Bacon·2023년 9월 27일
0

코딩테스트

목록 보기
3/45
post-thumbnail
post-custom-banner

문제

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

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

1열2열3열4열5열6열7열8열9열
1행32385341774255265
2행10739428852147263
3행874218785345188453
4행342864851216753655
5행21774535287590761
6행258765152811372874
7행65277541789786439
8행47477045236534144
9행871382383112292980

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

입력

첫째 줄부터 아홉 번째 줄까지 한 줄에 아홉 개씩 수가 주어진다. 주어지는 수는 100보다 작은 자연수 또는 0이다.

출력

첫째 줄에 최댓값을 출력하고, 둘째 줄에 최댓값이 위치한 행 번호와 열 번호를 빈칸을 사이에 두고 차례로 출력한다. 최댓값이 두 개 이상인 경우 그 중 한 곳의 위치를 출력한다.

예제 입력예제 출력
3 23 85 34 17 74 25 52 65
10 7 39 42 88 52 14 72 63
87 42 18 78 53 45 18 84 53
34 28 64 85 12 16 75 36 55
21 77 45 35 28 75 90 76 1
25 87 65 15 28 11 37 28 74
65 27 75 41 7 89 78 64 39
47 47 70 45 23 65 3 41 44
87 13 82 38 31 12 29 29 80
90
5 7

풀이

설계

2차원 배열에서 최댓값을 찾아야 하므로 단순하게 max를 사용하는 것으로는 해결이 어려울 것으로 판단, 각 행을 순회하면서 최댓값과 index를 저장하고 이를 행별로 비교하는 방법을 사용해보았다.
단계별로 풀면 다음과 같다.
1. 주어진 2차원 배열을 행별로 순회하면서 각 행의 최댓값을 찾는다.
2. 찾아낸 최댓값과 그 위치(행과 열)를 각각 저장한다. 이때 위치에서 행은 for문을 돌릴 때 사용할 i의 값을 저장하고, 열은 index()를 이용해 찾아 저장한다.
3. 행이 바뀔 때 이번 행의 최댓값을 이전 행의 최댓값과 비교하여 이전의 최댓값보다 클 경우 2의 3개 값을 대체한다.
4. 최종적으로 남은 값들을 출력한다.

코드

# 입력을 2차원 행렬로 받는다.
mat = [[0] * 9] * 9
for n in range(9):
    mat[n] = (list(map(int, input().split())))

# 첫 번째 행의 최댓값과 그 위치를 각각 저장한다.
max_val = max(mat[0])
row = 0
column = mat[0].index(max_val)

# 두 번째 행부터 마지막 행까지 순회하며 최댓값을 비교한다.
for i in range(1, 9):
	# 만약 새로운 최댓값이 이전의 최댓값보다 크다면 해당 값과 위치를 저장한다.
    if max(mat[i]) > max_val:
        max_val = max(mat[i])
        row = i
        column = mat[i].index(max_val)
    # 아님 말고
    else:
        pass

# 요구하는 출력값에 맞추어 행과 열 값에 1을 더해 출력한다.
print(max_val)
print(row + 1, column + 1)
profile
AI, NLP, Data analysis로 나아가고자 하는 개발자 지망생
post-custom-banner

0개의 댓글