[Lv. 0] 길이에 따른 연산

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/181879

프로그래머스

난이도 : Level.0

언어 : Kotlin

 

문제 설명 :

정수가 담긴 리스트 num_list가 주어질 때, 리스트의 길이가 11 이상이면 리스트에 있는 모든 원소의 합을 10 이하이면 모든 원소의 곱을 return하도록 solution 함수를 완성해주세요.

 

제한 사항 : 

• 2 ≤ num_list의 길이 ≤ 20
• 1 ≤ num_list의 원소 ≤ 9

입출력 예 :

num_list	result
[3, 4, 5, 2, 5, 4, 6, 7, 3, 7, 2, 2, 1]	51
[2, 3, 4, 5]	120

풀이 : 

class Solution {
    fun solution(num_list: IntArray): Int {
        return if (num_list.size > 10) {
            num_list.sumOf { it }
        } else {
            var temp = 1

            for (i in num_list.indices) {
                temp *= num_list[i]
            }

            temp
        }
    }
}

 

else의 for문을 줄인다면 어떻게 줄일 수 있을지 궁금했는데, 다른 사람들의 fold, reduce 함수 풀이법을 보고 이해했다.

해당 문제를 줄인다면 다음과 같이 줄일 수 있을 것이다.

 

class Solution {
    fun solution(num_list: IntArray): Int = if (num_list.size > 10) num_list.sumOf { it } else num_list.fold(1) { acc, i -> acc * i }
}