[Programmers] Lv1.신규 아이디 추천(kotlin)

SW Test/Programmers

[Programmers] Lv1.신규 아이디 추천(kotlin)

An effort will never betray 😎 2022. 7. 18. 22:43

문제

https://school.programmers.co.kr/learn/courses/30/lessons/72410?language=kotlin

예시

풀이

이 문제는 메소드 체이닝 방식으로 여러 번의 메소드를 호출하면 빠르게 풀 수 있습니다. 하지만 그만큼 다양한 함수에 대해서 알지 못하면 하나씩 로직을 작성하는 수밖에 없습니다.
해당 풀이는 문제에서 요구하는 대로 하나씩 맞춰가면서 풀 수 있습니다.

class Solution {
    fun solution(new_id: String): String {
        var answer: String = ""
        
        val strbuilder = StringBuilder()
        new_id.lowercase().forEach{
            if(it.isLetter() || it.isDigit() || it in arrayOf('-','_','.')){
                if(!(it == '.' && strbuilder.isNotEmpty() && strbuilder.get(strbuilder.length - 1) == '.')){
                    strbuilder.append(it)
                }
            }
        }
        
        if(strbuilder.get(0) == '.') strbuilder.deleteCharAt(0)
        if(strbuilder.isNotEmpty() && strbuilder.get(strbuilder.length - 1) == '.') strbuilder.deleteCharAt(strbuilder.length -1)
        if(strbuilder.isEmpty()) strbuilder.append('a')
        
        if(strbuilder.length >=16){
            val result = strbuilder.substring(0, 15)
            if(result.get(result.length - 1) == '.') return result.substring(0, 14)
            return result
        }else if(strbuilder.length == 2){
            strbuilder.append(strbuilder.get(strbuilder.length - 1))
        }else if(strbuilder.length == 1){
            strbuilder.append(strbuilder.get(strbuilder.length - 1))
            strbuilder.append(strbuilder.get(strbuilder.length - 1))
        }
        
        return strbuilder.toString()
    }
}

정규식을 사용하면 편하겠다는 생각을 했지만 방법을 몰라 별도로 공부를 했습니다. 일반적으로 코딩테스트에 통과하기 위한 수준으로 정규식을 배워야 하는지에 대한 의문은 있지만 알면 훨씬 빠르게 풀 수 있는 건 사실입니다.
아래 코드처럼 메소드 체이닝 기법을 사용하면 가독성 뿐만 아니라 코드를 정말 짧게 구현할 수 있습니다. 함수형 프로그래밍의 위대함을 또 한번 느끼네요

class Solution {
    fun solution(new_id: String): String = new_id.toLowerCase()
            .filter { it.isLetter() || it.isDigit() || it in arrayOf('-','_','.') }
            .replace("[.]+".toRegex(), ".")
            .removePrefix(".")
            .removeSuffix(".")
            .let { if(it.isEmpty()) "a" else it }
            .let { if(it.length >=16) it.substring(0, 15).removeSuffix(".") else it }
            .let {
                if(it.length == 1) it + it + it
                else if(it.length == 2) it + it.last()
                else it
            }
}