[프로그래머스/C++ 문제 풀이] Lv. 1 - 둘만의 암호

문제 설명

두 문자열 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

입출력 예

 

입출력 예 #1
본문 내용과 일치합니다.

---

문제 풀이

이번 문제는 skip 문자열에 포함된 문자를 제외하고 index만큼 모든 문자를 넘기며 특정 문자열을 찾는 문제입니다.

 

사실 처음에는 find() 함수를 사용해 특정 문자가 skip 문자열에 포함되어있는지를 체크하였으며 실제로 제한사항에 주어진 문자들의 길이가 적어 상관 없지만, 만약 문자열이 긴 경우 find()함수를 사용하는 것보다 26칸의 배열을 사용하는 것이 더욱 효율적이기에 이번에는 배열로 만들어 사용하게 되었습니다.

정답 코드

풀이 시간 : 10m 12s

#include <string>
#include <vector>

using namespace std;

int skipArr[26];

char Check(char c, int idx)
{      
    while(idx > 0)
    {
        // 문자 넘기기
        c++;
        
        while (true)
        {           
            // 해당 문자가 범위를 넘은 경우
            if (c > 'z')
            {
                // 문자 초기화
                c = 'a';
            }
            // 해당 문자가 금지 단어라면?
            else if (skipArr[c - 'a'] == 1)
            {
                // 하나 더 넘기기
                c++;
            }
            // 문제 없는 경우 종료
            else
            {
                break;
            }
        }
        
        // 인덱스 감소
        idx--;
    }
    
    return c;
}

string solution(string s, string skip, int index) {
    string answer = "";
    
    // 건너 뛸 문자열 추가
    for (char c : skip)
    {
        skipArr[c - 'a'] = 1;
    }
    
    // 전체 문자열 순회하며 정답 문자 추가하기
    for (char c : s)
    {
        answer += Check(c, index);
    }
    
    return answer;
}