[Programmers] OX 퀴즈

 

func solution(_ quiz:[String]) -> [String] {
    var result: [String] = []
    
    for i in quiz{
        var arr = i.components(separatedBy: " ")
        if arr[1] == "-"{
            if Int(arr[0])! - Int(arr[2])! == Int(arr[4])!{
                result.append("O")
            }
            else{
                result.append("X")
            }
        }
        else{
            if Int(arr[0])! + Int(arr[2])! == Int(arr[4])!{
                result.append("O")
            }
            else{
                result.append("X")
            }
        }
    }
    
    return result
}

 

프로그래머스