[백준 2476] 주사위 게임 (Swift)

2476번: 주사위 게임

• 세 주사위의 눈 중에 같은 눈이 무엇인지 찾아내는 것이 관건이었던 문제.
• 3개가 같을 때, 모두 다를 때는 조건문으로 어렵지 않게 처리 가능
• 문제는 2개가 같을 때인데
  • 여러 방법이 있을 거라고 생각했지만 내가 생각한 방법은 첫 번째 원소를 골라 그 원소와 같은 원소의 갯수가 2개이면 첫 번째 원소를, 그렇제 않으면 마지막 원소를 같은 눈으로 정했다.
  • 2개의 눈이 같은 경우는 3가지가 있다 : aab, aba, baa
  • 첫 번째 원소가 2개의 같은 눈에 포함되지 않는다면(aab가 아니라면), 나머지 경우를 보았을 때 반드시 세 번째 원소가 2개의 같은 눈에 해당되게 된다 (aba, baa)
• 가장 큰 상금을 찾는 것은 반복문 바깥에 상금 변수를 두고, 이번 회차에 계산한 상금이 기존 상금보다 클 때 대체하도록 했다.

풀이

let count = Int(readLine()!)!
var answer = 0
(1...count).forEach { _ in
    let inputArr = readLine()!.split(separator: " ").map { Int($0)! }
    let a = inputArr[0]
    let b = inputArr[1]
    let c = inputArr[2]
    var prize = 0
    if a == b && b == c && a == c {
        prize = 10000 + (a * 1000)
    } else if a == b || b == c || a == c {
        var equalNum = 0
        if inputArr.filter({ $0 == a }).count == 2 {
            equalNum = a
        } else {
            equalNum = c
        }
        prize = 1000 + equalNum * 100
    } else {
        prize = inputArr.max()! * 100
    }
    if prize > answer {
        answer = prize
    }
}
print(answer)