문제
출처: 2019 국가 교육기관 코딩 테스트
여러개의 숫자 카드 중에서 가장 높은 숫자가 쓰인 카드 한장을 뽑아야한다. 하지만, 게임의 룰을 지키며 카드를 뽑아야하고, 룰은 다음과 같다.
-
숫자가 쓰인 카드들이 N * M 형태로 놓여 있다.
이때 N은 행의 개수를 의미하며, M은 열의 개수를 의미한다. -
먼저 뽑고자 하는 카드가 포함되어 있는 행을 선택한다.
-
그다음 선택된 행에 포함된 카드들 중 가장 숫자가 낮은 카드를 뽑아야 한다.
-
따라서 처음에 카드를 골라낼 행을 선택할 때, 이후에 해당 행에서 가장 숫자가 낮은 카드를 뽑을 것을 고려하여 최종적으로 가장 높은 숫자의 카드를 뽑을 수 있도록 전략을 세워야 한다.
예를 들어 3 * 3 형태로 카드들이 다음과 같이 놓여 있다고 가정하자.
| 3 | 1 | 2 |
| 4 | 1 | 4 |
| 2 | 2 | 2 |
여기서 카드를 골라낼 행을 고를 때:
- 첫 번째 행을 선택하면 최종적으로 뽑는 카드는
1이다. - 두 번째 행을 선택해도 최종적으로 뽑는 카드는
1이다. - 그러나 세 번째 행을 선택하면 최종적으로 뽑는 카드는
2가 된다.
따라서 이 예제에서는 세 번째 행을 선택하여 숫자 2가 쓰여진 카드를 뽑는 것이 정답이다.
카드들이 (N \times M) 형태로 놓여 있을 때, 게임의 룰에 맞게 카드를 뽑는 프로그램을 작성하시오.
입력 조건
- 첫째 줄에 숫자 카드들이 놓인 행의 개수 N과 열의 개수 M이 공백을 기준으로 하여 각각 자연수로 주어진다. (1<=N, M<=100)
- 둘째 줄부터 (N)개의 줄에 걸쳐 각 카드에 적힌 숫자가 주어진다. 각 숫자는 1 이상 10,000 이하의 자연수이다.
출력 조건
- 첫째 줄에 게임의 룰에 맞게 선택한 카드에 적힌 숫자를 출력한다.
입력 예시 1
3 3
3 1 2
4 1 4
2 2 2출력 예시 1
2입력 예시 2
2 4
7 3 1 8
3 3 3 4출력 예시 2
3해답 코드 · 주요 로직 설명
문제를 보고 각 행마다 가장 작은 수들을 뽑아내고, 해당 수 중에서 가장 큰 수를 출력시키면 된다고 생각하였다.
- 처음 작성한 코드
N, M = map(int, input().split())
max = 0
for i in range(N):
data = list(map(int, input().split()))
if max == 0:
max = min(data)
if min(data) > max:
max = min(data)
print(max)- 개선 코드
if문으로 조건을 거는 것 보다 min, max 함수를 사용해서 최소, 최대 값을 뽑아내는 것이 더 짧은 코드 런타임을 가져갈 수 있어 해당 부분을 수정했다.
N, M = map(int, input().split())
max_value = 0
for i in range(N):
data = list(map(int, input().split()))
max_value = max(max_value, min(data))
print(max_value) 해당 글은 다음 도서의 내용을 정리하고 참고한 글임을 밝힙니다. 보다 자세한 내용은 아래 책에서 확인할 수 있습니다. 나동빈, ⌜이것이 취업을 위한 코딩 테스트다 with 파이썬⌟, 한빛미디어, 2020, 604쪽
IT, 개발 관련 정보들을 기록하는 장소입니다.