본문 바로가기
Back-End/Kotlin

[Kotlin]가변인자(Vararg) 사용하기

by codeZeroMan 2023. 7. 10.

가변인자 : Vararg(Variable number of arguments)

가변인자란?

  • 함수의 인자의 갯수를 정하지 않고 동적으로 받는것을 말한다.

예제를 통한 사용법

예제1. 정해지진 않은 갯수의 수를 입력받아 총합을 구하는 함수를 작성하시오

fun sumVararg(vararg numbers : Int):Int{ 
    var sum=0
    numbers.forEach{
        n-> sum=sum+n
    }
    return sum
}

예제2. 정해지진 않은 갯수의 수와 사칙연산을 입력받아 계산하는 함수를 작성하시오

fun sumVararg(vararg numbers: Int, arithmeticOperation: Char): Int {
    var sum = 0
    numbers.forEach { n ->
        when (arithmeticOperation) {
            '+' -> sum += n
            '-' -> sum -= n
            '/' -> sum /= n
            '*' -> sum *= n
            else -> throw RuntimeException("Invalid arithmetic operation: $arithmeticOperation")
        }
    }
    return sum
}