그리디 : 숫자 카드 게임 [ 이것이 코딩테스트다 ] [ Python ]

문제

숫자 카드 게임은 여러 개의 숫자 카드 중에서 가장 높은 숫자가 쓰인 카드 한 장을 뽑는 게임이다.
단, 게임의 룰을 지키며 카드를 뽑아야 하고 룰은 다음과 같다.

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

    N
 +  +  +
[3, 1, 2] +
[4, 1, 4] +  M
[2, 2, 2] +

여기서 카드를 골라낼 행을 고를 때 첫 번째 혹은 두 번째 행을 선택하는 경우, 최종적으로 뽑는
카드는 1 이다. 하지만 세 번째 행을 선택하는 경우 최종적으로 뽑는 카드는 2 이다.
따라서 이 예제에서는 세 번째 행을 선택하여 숫자 2 가 쓰여진 카드를 뽑는 것이 정답이다.

카드들이 N * 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

tip ) 각 행마다 가장 작은 수를 찾은 뒤에 그 수 중에서 가장 큰 수를 찾는 것

---

코드

n, m = map(int, input().split())
# n 변수와 m 변수를 입력받습니다.

result = 0

for i in range(n):
    # n 의 크기 만큼 루프를 반복합니다.
    arr = list(map(int, input().split()))  
    # n 만큼 값을 입력받는다.
    minNum = min(arr)
    # arr 중에 최소값을 minNum 변수에 넣습니다.
    result = max(minNum, result)
    # 큰 값을 result 에 저장합니다.
    
print(result)

 

저자 코드

# 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) # 최종 답안 출력

---

회고

이 문제에서 입력 조건에 대한 것을 생각하지 못하고 풀었던 문제가 있었다. 입력 조건 포함해 문제를 잘봐야할 것 같다.

파이썬은 내장 함수가 활용하기 좋은 것 같다. 적절한 때에 내장 함수를 잘 활용한다면 더 좋은 코드가 완성될 것 같다.