[백준 4493] 가위 바위 보? (Swift)

4493번: 가위 바위 보?

• 맨 첫 줄에는 테스트 케이스의 갯수가 주어진다
• 그 아래부터는 테스트 케이스가 주어진다
  • 테스트 케이스의 첫 줄은 가위바위보 게임 횟수가 주어진다
  • 그 아래부터는 가위바위보 게임 횟수만큼 무엇을 냈는지 주어진다
    • 왼쪽은 Player1, 오른쪽은 Player2
    • R = Rock = 바위, S = Scissor = 가위, P = Paper = 보
• 테스트 케이스마다 누가 이겼는지(또는 비겼는지) 출력한다

풀이

let count = Int(readLine()!)!
(1...count).forEach { num in
    let game = Int(readLine()!)!
    var score1 = 0
    var score2 = 0
    (1...game).forEach { _ in
        let inputArr = readLine()!.split(separator: " ")
        if inputArr.contains("R") && inputArr.contains("S") { // O(n)..
            if inputArr.first! == "R" {
                score1 += 1
            } else {
                score2 += 1
            }
        }
        if inputArr.contains("S") && inputArr.contains("P") {
            if inputArr.first! == "S" {
                score1 += 1
            } else {
                score2 += 1
            }
        }
        if inputArr.contains("P") && inputArr.contains("R") {
            if inputArr.first! == "P" {
                score1 += 1
            } else {
                score2 += 1
            }
        }
    }
    print(score1, score2)
    if score1 > score2 {
        print("Player 1")
    } else if score1 < score2 {
        print("Player 2")
    } else {
        print("TIE")
    }
}

Player1이 낸 패에 대해 Player2가 낸 패가 이기는지/비기는지/지는지 를 비교할 수도 있었지만,

나는 이기고 지는 패의 조합 (가위, 바위 / 바위, 보 / 가위, 보) 에 대해서만 비교를 하는 풀이가 먼저 떠올라 이렇게 구현해 보았다.

이런 경우 비길 때는 어떻게 스코어를 정하느냐?? 사실 생각지 못한 부분이었고 지금 풀이를 적으면서 놓쳤음을 깨달았지만..🥲 어차피 비겼을 경우에는 둘 중 아무 쪽에도 점수를 더하지 않으므로, 승패의 결과는 동일하다.

(채점을 해 보니 테스트 케이스가 많은 것 같진 않아서 아쉽다)

시간복잡도는 60ms가 나왔다.

보완점

if inputArr.contains("R") && inputArr.contains("S") { ... }

이 부분은 contains 함수가 O(n)의 시간복잡도를 가지므로,

(알고리즘 문제이니 입력이 일정하다는 가정 하에) 배열의 인덱스 접근으로 풀었다면 더 좋았을 것 같다.

let count = Int(readLine()!)!
(1...count).forEach { num in
    let game = Int(readLine()!)!
    var score1 = 0
    var score2 = 0
    (1...game).forEach { _ in
        let inputArr = readLine()!.split(separator: " ")
        if inputArr[0] == "R" && inputArr[1] == "S"
                        || inputArr[0] == "S" && inputArr[1] == "R" {
            if inputArr.first! == "R" {
                score1 += 1
            } else {
                score2 += 1
            }
        }
        if inputArr[0] == "S" && inputArr[1] == "P"
                        || inputArr[0] == "P" && inputArr[1] == "S" {
            if inputArr.first! == "S" {
                score1 += 1
            } else {
                score2 += 1
            }
        }
        if inputArr[0] == "P" && inputArr[1] == "R"
                        || inputArr[0] == "R" && inputArr[1] == "P" {
            if inputArr.first! == "P" {
                score1 += 1
            } else {
                score2 += 1
            }
        }
    }
    if score1 > score2 {
        print("Player 1")
    } else if score1 < score2 {
        print("Player 2")
    } else {
        print("TIE")
    }
}

조건절 부분이 두배로 길어지긴 했지만 수정해 보았다 (결과는 동일하다)

시간복잡도는 이쪽이 56ms로 조금 더 빠르게 나왔다! 🙂