문제

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

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

수도 코드

1. skip 문자열을 문자 단위로 unordered_set에 저장 → skip_set
    
    2. 결과 문자열 answer 초기화

    3. 문자열 s의 각 문자 c에 대해 반복:
        a. ch = c (현재 문자에서 시작)
        b. count = 0 (유효 이동 횟수)

        c. while count < index:
            - ch = ch + 1 (알파벳 다음 문자로 이동)
            - 만약 ch > 'z' 이면, ch = 'a' (알파벳 순환)

            - 만약 ch가 skip_set에 포함되어 있으면:
                - continue (건너뛰고 count는 증가시키지 않음)

            - count += 1 (유효한 이동으로 인정)

        d. answer에 ch 추가

    4. answer 반환

코드

#include<bits/stdc++.h>
using namespace std;
string solution(int index, string skip, string s){
    unordered_set<char> skip_set(skip.begin(), skip.end());
    string answer = "";
    for(auto c : s){
        char ch = c;
        int count = 0;

        while(count < index){
            ch++;
            if(ch > 'z') ch = 'a';
            if(skip_set.count(ch)){
                continue;
            }
            count++;
        }
        answer += ch;
    }
    return answer;
}

Reference

나의 뇌