숫자 카드 게임

이것이 취업을 위한 코딩 테스트다 with 파이썬을 공부하면서 정리한 내용입니다.

---

문제 내용

• 숫자 카드 게임은 여러 개의 숫자 카드 중에서 가장 높은 숫자가 쓰인 카드 한장을 뽑는 게임
• 게임의 룰을 지키며 카드를 뽑아야 하고 룰은 다음과 같음
  1. 숫자가 쓰인 카드들이 $N \times M$ 형태로 놓여 있으며 이때 N은 행의 개수, M은 열의 개수를 의미
  2. 면저 뽑고자 하는 카드가 포함되어 있는 행을 선택
  3. 그다음 선택된 행에 포함된 카드들 중 가장 숫자가 낮은 카드를 뽑음
  4. 따라서 처음에 카드를 골라낼 행을 선택할 때, 이후에 해당 행에서 가장 숫자가 낮은 카드를 뽑을 것을 고려하여 최종적으로 가장 높은 숫자의 카드를 뽑을 수 있도록 전략을 세워야 함

입력 조건

• 첫째 줄에 숫자 카드들이 놓인 행의 개수 N과 열의 개수 M이 공백을 기준으로 하여 각각 자연수로 주어짐
• 둘째 줄부터 N개의 줄에 걸쳐 각 카드에 적힌 숫자가 주어짐

출력 조건

• 첫째 줄에 게임의 룰에 맞게 선택한 카드에 적힌 숫자를 출력

입력 예시 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

---

문제 해설

• 각 행마다 가장 작은 수를 찾은 뒤에 그 수 중에서 가장 큰 수 찾으면 됨
• 리스트에서 가장 작은 원소를 찾아주는 min() 함수를 이용

소스 코드

# N, M을 공백으로 구분하여 입력받기
n, m = map(int, input().split())

result = 0
# 한 줄씩 입력받아 계산
for i in range(n):
    data = list(map(int, input().split()))
    # 현재 줄에서 '가장 작은 수' 찾기
    min_value = min(data)
    # '가장 작은 수'들 중에서 가장 큰 수 찾기
    result = max(result, min_value)

print(result) # 최종 답안 출력

다른 풀이

• 2중 반복문 구조를 이용

소스 코드

# N, M을 공백으로 구분하여 입력받기
n, m = map(int, input().split())

result = 0
# 한 줄씩 입력받아 계산
for i in range(n):
    data = list(map(int, input().split()))
    # 현재 줄에서 '가장 작은 수' 찾기
    min_value = 10001
    for a in data:
        min_value = min(min_value, a)
    # '가장 작은 수'들 중에서 가장 큰 수 찾기
    result = max(result, min_value)

print(result) # 최종 답안 출력

---

나의 풀이

• 리스트 컴프리헨션을 사용하여 숫자 카드들이 담긴 2차원 리스트 생성
• 반복문과 min() 메서드를 이용해 2차원 리스트 안에 있는 1차원 리스트들 각각에서 가장 작은 숫자 찾기
• 가장 작은 숫자들로 리스트를 만들고 해당 리스트에서 max() 메서드로 가장 큰 수 찾기

소스 코드

import sys

# n은 행의 개수, m은 열의 개수
n, m = map(int, sys.stdin.readline().rstrip().split())
# 숫자 카드를 저장할 2차원 리스트
matrix = [list(map(int, sys.stdin.readline().rstrip().split())) for _ in range(n)]

# 각 행에서 가장 작은 숫자를 찾음
min_list = [min(array) for array in matrix]
# 작은 숫자들 중에서 가장 큰 것을 선택
result = max(min_list)

print(result)

다른 풀이

• 2차원 리스트를 만들고 그 안에서 가장 작은 숫자들의 리스트를 다시 만드는 것 대신 처음 입력받을 때부터 min() 메서드 사용하여 가장 작은 숫자들로 이루어진 리스트 생성
• 가장 작은 숫자들로 만들어진 리스트에서 max() 메서드로 가장 큰 수 찾기

소스 코드

import sys

# n은 행의 개수, m은 열의 개수
n, m = map(int, sys.stdin.readline().rstrip().split())

# 각 행에서 가장 작은 숫자를 저장
min_list = [min(list(map(int, sys.stdin.readline().rstrip().split()))) for _ in range(n)]

# 작은 숫자들 중에서 가장 큰 것을 선택
result = max(min_list)

print(result)