[프로그래머스_Kotlin] 둘만의 암호

프로그래머스

 

 

문제

• 두 문자열 s와 skip, 그리고 자연수 index가 주어질 때, 다음 규칙에 따라 문자열을 만들려 합니다. 암호의 규칙은 다음과 같습니다.
  
  • 문자열 s의 각 알파벳을 index만큼 뒤의 알파벳으로 바꿔줍니다.
  • index만큼의 뒤의 알파벳이 z를 넘어갈 경우 다시 a로 돌아갑니다.
  • skip에 있는 알파벳은 제외하고 건너뜁니다.
• 예를 들어 s = "aukks", skip = "wbqd", index = 5일 때, a에서 5만큼 뒤에 있는 알파벳은 f지만 [b, c, d, e, f]에서 'b'와 'd'는 skip에 포함되므로 세지 않습니다. 따라서 'b', 'd'를 제외하고 'a'에서 5만큼 뒤에 있는 알파벳은 [c, e, f, g, h] 순서에 의해 'h'가 됩니다. 나머지 "ukks" 또한 위 규칙대로 바꾸면 "appy"가 되며 결과는 "happy"가 됩니다.
  
  
• 두 문자열 s와 skip, 그리고 자연수 index가 매개변수로 주어질 때 위 규칙대로 s를 변환한 결과를 return하도록 solution 함수를 완성해주세요.

 

제한사항

• 5 ≤ s의 길이 ≤ 50
• 1 ≤ skip의 길이 ≤ 10
• s와 skip은 알파벳 소문자로만 이루어져 있습니다.
  
  • skip에 포함되는 알파벳은 s에 포함되지 않습니다.
• 1 ≤ index ≤ 20

 

입출력 예

 

 

 

내 풀이

class Solution {
    fun solution(s: String, skip: String, index: Int): String {
        var answer: String = ""

        s.map {
            var cnt = index
            var str = it

            while (cnt > 0) {
                str++
                if (str > 'z') str = 'a'
                if (!skip.contains(str)) cnt--
            }
            answer += str
        }
        return answer
    }
}

 

 

해석

s.map { ... }

 

1. s.map으로 문자열 한 개씩 순회

var cnt = index
var str = it

2. index와 s의 문자열을 새로 받음

while (cnt > 0) {
	str++
	if (str > 'z') str = 'a'
	if (!skip.contains(str)) cnt--
}

3. index만큼 알파벳을 넘겨야하는데 skip에 포함되면 카운트하지말고 넘겨야해서 while문으로 생성

str++로 s의 문자열 하나씩 올리는데 if문 2개를 생성해줬다.

먼저 z초과면 a로 돌리기

마지막으로 skip에 포함되지 않으면 카운트 감소

 

 

다른사람의 풀이

class Solution {
    fun solution(s: String, skip: String, index: Int): String {
        var answer: String = ""
        val skipped = skip.map { it.toChar() }
        val alphabet = ('a'..'z').filter { it !in skipped }

        s.forEach {
            val i = (alphabet.indexOf(it) + index) % alphabet.size
            answer += alphabet[i]
        }
        return answer
    }
}

 

skipped : 문자열 한개씩 저장

alphabet : skipped를 제외한 알파벳 나열

s를 forEach로 문자열 각각 입력받는다.

alphabet.indexOf(it) : alphabet의 문자열의 위치에 index만큼 더하고 % alphabet.size로 나머지를 구한다.

'z'를 초과할까봐 한 것이다.

answer에 값을 입력한다.