[Algorithm] 프로그래머스 - 점의 위치 구하기 (Kotlin)

사분면은 한 평면을 x축, y축을 기준으로 나눈 네 부분이다. x좌표와 y좌표 (x, y)를
차례대로 담은 정수 배열 dot이 매개변수로 주어진다. 좌표 dot이 사분면 중 어디에
속하는지 1, 2, 3, 4를 리턴하는 solution()을 완성하라

 

 

조건문을 쓰면 간단하게 구할 수 있다. if를 쓴다면 아래와 같이 할 수 있다.

 

class Solution {
    fun solution(dot: IntArray): Int = if (dot[0] > 0 && dot[1] > 0) {
        1
    } else if (dot[0] > 0 && dot[1] < 0) {
        4
    } else if (dot[0] < 0 && dot[1] > 0) {
        2
    } else {
        3
    }
}

 

when을 쓴다면 아래와 같다.

 

class Solution {
    fun solution(dot: IntArray): Int = when {
        (dot[0] > 0 && dot[1] > 0) -> 1
        (dot[0] > 0 && dot[1] < 0) -> 4
        (dot[0] < 0 && dot[1] > 0) -> 2
        else -> 3
    }
}