[2566] 백준 문제풀이 - 최댓값

https://www.acmicpc.net/problem/2566

문제

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

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

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

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

입력

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

출력

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

Python Code

백준 문제를 풀 때에는 항상 반례를 고민하며 코드를 짜야한다. 만약 반복문이 0부터 시작될 경우에 모든 열과 행의 수가 0이라면 인덱스가 0,0으로 나와 문제풀이에 실패하게 된다. 반복문을 1부터 실행시켜 인덱스 초기값을 1,1로 줘버리면 해결할 수 있다.

table = []
for _ in range(9):
    num = list(map(int, input().split(" ")))
    table.append(num)

max_num = 0
max_width, max_height = 1, 1
for i in range(1, 10):
    if max(table[i - 1]) > max_num:
        max_num = max(table[i - 1])
        max_width = i
        max_height = table[i - 1].index(max_num) + 1
print(max_num)
print(max_width, max_height)