[프로그래머스] 7의 개수 - Kotlin

 

 

 

 

 

문제 설명

머쓱이는 행운의 숫자 7을 가장 좋아합니다. 정수 배열 array가 매개변수로 주어질 때, 7이 총 몇 개 있는지 return 하도록 solution 함수를 완성해보세요.

 

 

 

 

 

제한사항

• 1 ≤ array의 길이 ≤ 100
• 0 ≤ array의 원소 ≤ 100,000

 

 

 

 

입출력 예

 

 

 

 

 

입출력 예 설명

입출력 예 #1

• [7, 77, 17]에는 7이 4개 있으므로 4를 return 합니다.

입출력 예 #2

• [10, 29]에는 7이 없으므로 0을 return 합니다.

 

 

 

 

나의 풀이

class Solution {
    fun solution(array: IntArray): Int {
        var answer: Int = 0

        array.map {
            it.toString().split("").map {
                if(it == "7") {
                    answer++;
                }
            }
        }

        return answer
    }
}

 

 

 

 

 

 

AI 리팩토링 코드

class Solution {
    fun solution(array: IntArray): Int {
        var answer = 0
        array.forEach {
            it.toString().forEach { c ->
                if (c == '7') {
                    answer++
                }
            }
        }
        return answer
    }
}

 

 map() 함수는 주어진 함수를 모든 요소에 대해 적용하고 새로운 리스트를 반환하는데, 이 경우에는 새로운 리스트를 반환할 필요가 없습니다. 따라서 forEach() 함수를 사용하면 더욱 효율적인 코드를 작성할 수 있습니다.