[Kotlin] 시저 암호 (프로그래머스 알고리즘)

 

 

 

 

문제

https://school.programmers.co.kr/learn/courses/30/lessons/12926

 

 

 

 

---

 

접근

 

• 아스키 코드를 이용해서 풀면 되지 않을까? 생각하며 아래와 같이 구상했다.

대충 구상한 코드 
class Solution {
    fun solution(s: String, n: Int): String {
        val temp = s.split("")
        var answer = temp.map { it.toInt() + n }
        return answer.toString()
    }
}

• 위 코드는 java.lang.NumberFormatException: For input string: "" 오류가 뜨며 작동하지 않는다.
• NumberFormatException이 발생하는 이유는 s.split("")를 통해 문자열을 빈 문자열("")을 기준으로 분할했기 때문이다. 예를 들어, 문자열이 "abc"라면 s.split("")는 ["", "a", "b", "c", ""]와 같이 배열을 반환한다. 여기서 빈 문자열이 포함되어 있기 때문에, toInt() 메서드를 호출할 때 빈 문자열("")로 인해 NumberFormatException이 발생한다.
• 따라서 문자열을 분할할 때 빈 문자열을 기준으로 하지 않고, 각 문자를 분할하기 위해 다른 방법을 사용해야 한다.
  문자열을 각 문자 단위로 분할하는 방법으로는 toCharArray() 메서드를 사용할 수 있다.

 

class Solution {
    fun solution(s: String, n: Int): String {
        val answer = s.toCharArray().map { (it.toInt() + n).toChar() }.joinToString("")
        return answer
    }
}

• 그래서 이런식으로 수정하였는데, 이번엔 테스트케이스가 문제였다. 
  오직 "AB" "1" -> "BC" 와 같은 경우에만 동작하고, z에서 a로 가거나 띄어쓰기를 반영하지는 못했다.

 

 

 

---

 

성공 코드

 

class Solution {
    fun solution(s: String, n: Int): String {
        return s.toCharArray().map{
            when(it){
                in 'A'..'Z' -> {
                    if(it + n > 'Z') it + n - 26 else it + n
                }in 'a'..'z' -> {
                    if(it + n > 'z') it + n - 26 else it + n
                }else -> {
                    it
                }
            }
        }.joinToString("")
    }
}

 

 

 

---

 

설명

 

1. 시저 암호에서 중요한 것은 내가 처음 구상했던 것처럼 단순히 아스키 코드값에서 n값(거리)를 더하지 않는 것이다. 그렇게 구성하면 알파벳의 범위를 벗어나서 이상한 값이 출력될 경우의 수가 있기 때문이다. 따라서 26을 이용해서 알파벳을 벗어날 수 없도록 만드는 것이 중요하다.
2. string이었던 s를 char요소의 배열로 만들고, 각각의 요소를 map으로 처리한 뒤 joinToString("")으로 반환한다.
3. A..Z 대문자에서 대문자를 넘어가는 경우 더한 값에서 26을 빼준다.
   a..z 소문자에서도 마찬가지로 소문자를 넘어가는 경우 더한 값에서 26을 빼준다.
   공백을 처리하기 위해 else는 it을 그대로 반환한다. 
4. -26을 하는 이유는 알파벳의 순환성 때문이다.
   알파벳은 순환 구조를 가지고 있어서 'Z' 다음에는 다시 'A'가 온다. 따라서 알파벳을 오른쪽으로 n만큼 이동시킬 때, 'Z'를 넘어가는 경우 다시 'A'로 돌아와야 한다. 'Z'의 ASCII 값은 90이고, 'A'의 ASCII 값은 65이다. 따라서 'Z'에 1을 더하면 91이 되는데, 다시 처음부터 시작하려면 -26을 해주어야 한다.
   따라서 'Z'에서 오른쪽으로 1만큼 이동한 결과는 90 + 1 - 26 = 65가 된다.

 

 

 

---

 

다른 방법

 

class Solution {
    fun solution(s: String, n: Int): String =
        s.toList().joinToString(separator = "") {
            when (it) {
                in 'A'..'Z' -> ('A'.toInt() + (it.toInt() - 'A'.toInt() + n) % ('Z' - 'A' + 1)).toChar()
                in 'a'..'z' -> ('a'.toInt() + (it.toInt() - 'a'.toInt() + n) % ('z' - 'a' + 1)).toChar()
                else -> it
            }.toString()
        }
}

• A와 Z의 아스키코드차가 26인지 알지 못해도 사용할 수 있는 방법이다.

 

class Solution {
    fun solution(s: String, n: Int): String =
        s.map {
            when {
                it.isLowerCase() -> 'a' + (it + n - 'a') % 26
                it.isUpperCase() -> 'A' + (it + n - 'A') % 26
                else -> ' '
            }
        }.joinToString("")
}

• 'a'..'z' 와 같이 표기하지 않고, isLowerCase()와 isUpperCase()를 이용한 방법이다.