코딩테스트/문제풀이

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

알렉스 페레이라 2024. 10. 24. 11:08

알파벳 넘어가는것은 그냥 if(index>25){idex-=26;} 하면된다.. 짧게 잘 푼듯


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


내 코드

class Solution {
    public String solution(String s, String skip, int index) {
        String answer = "";
        
        for(int i = 0; i < s.length(); i++){
            answer += getAlphabet(String.valueOf(s.charAt(i)), skip, index);
        }        
        return answer;
    }
    
    public String getAlphabet(String s, String skip, int index){
        String ab = "abcdefghijklmnopqrstuvwxyz";
        int start = ab.indexOf(s); //시작 위치
        
        while(index > 0){
            start++; //한칸 이동
            if(start > 25)start-=26; //z를 넘어서면 a로
            if(skip.indexOf(String.valueOf(ab.charAt(start))) == -1){//다음 알파벳이 skip에 없을때만 횟수 차감
                index--;
            }
        }
        return String.valueOf(ab.charAt(start));
        
    }
}