[프로그래머스] 문자열 암호화 문제 풀기 with golang

문제 설명
두 문자열 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 함수를 완성해주세요.

 

func solution(s string, skip string, index int) string {
	result := ""
	skipSet := make(map[rune]bool)

	for _, c := range skip {
		skipSet[c] = true
	}

	for _, c := range s {
		nextChar := c
		skipcount := 0

		for i := 0; i < index+skipcount; i++ {
			nextChar++

			if nextChar > 'z' {
				nextChar = 'a'
			}

			if skipSet[nextChar] {
				skipcount++
			}
		}
		result += string(nextChar)
	}

	return result
}

 

코드 설명
result는 변환된 문자열을 저장하기 위한 변수입니다.
skipSet은 skip 문자열에 포함된 문자를 빠르게 찾기 위한 map 자료구조입니다.
skip에 포함된 문자를 skipSet에 저장합니다.
문자열 s의 각 문자에 대해서 다음을 수행합니다.
nextChar에 현재 문자를 저장합니다.
skipcount는 건너뛸 문자의 개수를 저장하기 위한 변수입니다.
index와 skipcount를 합한 만큼 반복문을 돌면서 다음 문자를 찾습니다.
찾은 문자를 result에 추가합니다.
최종적으로 result를 반환하면 문제의 조건에 맞는 문자열 변환이 완료됩니다.