[Kotlin] Programmers 코딩 기초 트레이닝 Day 04 연산, 조건문

n의 배수

class Solution {
    fun solution(num: Int, n: Int): Int = if (num % n == 0) 1 else 0
}

 

 

 

공배수

class Solution {
    fun solution(number: Int, n: Int, m: Int): Int = if (number % n == 0 && number % m == 0) 1 else 0
}

 

 

 

홀짝에 따라 다른 값 반환하기

import kotlin.math.pow
class Solution {
    fun solution(n: Int): Int = (2 - (n % 2) .. n step 2).sumOf { (it.toFloat()).pow(2 - (n % 2)).toInt() }

}

 

 

 

조건 문자열

class Solution {
    fun solution(ineq: String, eq: String, n: Int, m: Int): Int {
        when(ineq){
            "<" -> {
                when(eq){
                    "=" -> if (n <= m) return 1
                    "!" -> if (n < m) return 1
                }
            }
            ">" -> {
                when(eq){
                    "=" -> if (n >= m) return 1
                    "!" -> if (n > m) return 1
                }
            }
        }
        return 0
    }
}

 

 

 

flag에 따라 다른 값 반환하기

class Solution {
    fun solution(a: Int, b: Int, flag: Boolean): Int = if (flag) a + b else a - b
}