Algorithm/[Swift] 이것이 코딩테스트다
[Greedy] 숫자 카드 게임
내일은개발천재🎵
2022. 12. 29. 13:12
문제
숫자 카드 게임은 여러 개의 숫자 카드 중에서 가장 높은 숫자가 쓰인 카드 한 장을 뽑는 게임이다. 단, 게임의 룰을 지키며 카드를 뽑아야 하고, 룰은 다음과 같다.
- 숫자가 쓰인 카드들이 N * M 형태로 놓여있다. 이때 N은 행의 개수를 의미하며, M은 열의 개수를 의미한다.
- 먼저 뽑고자 하는 카드가 포함되어 있는 행을 선택한다.
- 그 다음 선택된 행에 포함된 카드들 중 가장 숫자가 낮은 카드를 뽑아야 한다.
- 따라서 처음에 카드를 골라낼 행을 선택할 때, 이후에 해당 행에서 가장 숫자가 낮은 카드를 뽑을 것을 고려하여 최종적으로 가장 높은 숫자의 카드를 뽑을 수 있도록 전략을 세워야 한다.
예를 들어 3 * 3 형태로 카드르이 다음과 같이 놓여 있다고 가정하자.
3 1 2
4 1 4
2 2 2
여기서 카드를 골라낼 행을 고를 때 첫 번째 혹은 두 번째 행을 선택하는 경우, 최종적으로 뽑는 카드는 1이다. 하지만 세 번째 행을 선택하는 경우 최종적으로 뽑는 카드는 2이다. 따라서 이 예제에서는 세 번째 행을 선택하여 숫자 2가 쓰여진 카드를 뽑는 것이 정답니다.
코드
import Foundation
let input = readLine()!.split(separator: " ").map { Int($0)! }
var minNumbers = [Int]()
for _ in 0..<input[0] {
let colum = readLine()!.split(separator: " ").map { Int($0)! }
minNumbers.append(colum.min()!)
}
print(minNumbers.max()!)
알고리즘
- 입력받은 수들 중 가장 작은 값만 list에 저장한 후, 그 중 가장 큰 숫자를 출력한다.
분석
- O( n log n )
- for 문 -> O(n) , min max -> O(n log n)