본문 바로가기

프로그래머스 (Swift)/기초

[Day4-4] 조건 문자열

문제

 

내 정답 코드

import Foundation

func solution(_ ineq:String, _ eq:String, _ n:Int, _ m:Int) -> Int {
    
    if ineq == "<" {
        if eq == "=" {
            return n <= m ? 1 : 0
        } else {
            return n < m ? 1 : 0
        }
    } else {
        if eq == "=" {
            return n >= m ? 1 : 0
        } else {
            return n > m ? 1 : 0
        }
    }
}

#

1. 문자열을 비교하여 삼항연산자를 통해 해결했다.

 

문자열을 더하고 switch문을 사용하면 더 깔끔하게 해결 가능하다.

import Foundation

func solution(_ ineq:String, _ eq:String, _ n:Int, _ m:Int) -> Int {
    
    switch ineq + eq {
    case "<=":
        return n <= m ? 1 : 0
    case "<!":
        return n < m ? 1 : 0
    case ">=":
        return n >= m ? 1 : 0
    case ">!":
        return n > m ? 1 : 0
    default:
        return 0
    }
}

배운 기술