[백준 2484] 주사위 네개 (Swift)

2484번: 주사위 네개

• 2476 주사위 게임 과 비슷하게 풀면 되겠다고 생각했는데, 경우의 수가 훨씬 더 많았다.
• filter를 활용해 나온 주사위 눈을 모두 탐색해서 같은 눈이 몇 개인지 구한 뒤, 그에 따라 상금을 계산했다
  • 같은 눈이 몇 개인지 구하면서, 같은 눈인 숫자가 무엇인지도 저장함
• 같은 눈이 2개인 경우, 이미 찾은 같은 눈 2개를 제외한 나머지 배열(길이=2)의 값이 서로 같으면 2쌍으로 판단하고, 그렇지 않은 경우 한 쌍으로 판단했다.

풀이

let count = Int(readLine()!)!
var answer = 0
(1...count).forEach { _ in
    let inputArr = readLine()!.split(separator: " ").map { Int($0)! }
    var prize = 0
    var equalCount = 0
    var equalNum = 0
    for index in 0..<4 {
        let count = inputArr.filter { $0 == inputArr[index] }.count
        if count > equalCount {
            equalCount = count
            equalNum = inputArr[index]
        }
    }
    switch equalCount {
    case 4:
        prize = 50000 + equalNum * 5000
    case 3:
        prize = 10000 + equalNum * 1000
    case 2:
        let partialArr = inputArr.filter { $0 != equalNum }
        if partialArr[0] == partialArr[1] {
            let anotherEqualNum = partialArr[0]
            prize = 2000 + equalNum * 500 + anotherEqualNum * 500
        } else {
            prize = 1000 + equalNum * 100
        }
    case 1:
        prize = inputArr.max()! * 100
    default:
        return
    }
    if prize > answer {
        answer = prize
    }
}
print(answer)