[Algorithm] OX 문제

|  OX 문제

 

프로그래머스

 

 

import Foundation

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

 

 

• 함수는 [String] 타입의 quiz 배열을 입력 받는다.
• 각 문자열 q는 공백 문자로 구분하고, 이를 components(separatedBy:) 메서드를 사용하여 배열 list로 변환
• 배열 list의 두 번째 요소가 -인 경우, 첫 번째 숫자에서 세 번째 숫자를 뺀 값이 네 번째 숫자와 일치하는지 확인
• 배열 list의 두 번째 요소가 +인 경우, 첫 번째 숫자와 세 번째 숫자를 더한 값이 네 번째 숫자와 일치하는지 확인
• 연산 결과가 일치하면 "O"를, 일치하지 않으면 "X"를 yeon 배열에 추가
• 최종적으로 yeon 배열을 반환