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

문제 설명

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

Pseudo Code

skip 문자열 -> set 객체로
for s 문자열
  문자(c)가 set에 있으면 index+1만큼 증가 (c-'a'+index)%26+'a'
  없으면 index 수만큼 증가

 

작성한 코드

class Solution { 
  public String solution(String s, String skip, int index) { 
    StringBuilder answer = new StringBuilder(); 
    Set<Character> set = new HashSet<>(); 
    for(char c: skip.toCharArray()){ set.add(c); } 
    for(char sc : s.toCharArray()){ 
      for(int i: set){ 
        if(sc -'a' < i-'a' && (sc -'a'+index) >= i-'a'){ 
          answer.append((char)((sc -'a'+index+1)%26+'a')); 
        } else { 
          answer.append((char)((sc -'a'+index)%26+'a')); 
         }} 
       } 
       return answer.toString(); 
       } }

• 아스키코드로 접근하다가 skip 문자처리가 적용이 안됬다. 매 문자의 이동량을 index로 쓰는 것은 맞지만 skip에 몇개가 있을지 모름 그것까지 생각을 못함
• 자바는 char끼리 더하기연산(++)을 하면 자동으로 아스키 코드 값이 1 증가한다는 사실을 알게되었다. 예를들어 'a'++ -> 'b'
• 소문자만 필요하니 'z'에서 'a'로 넘어가는 조건문만 추가해서 문자 순환을 처리해보았다.

수정한 코드

class Solution {
    public String solution(String s, String skip, int index) {
        StringBuilder answer = new StringBuilder();
        Set<Character> set = new HashSet<>();
        for(char c: skip.toCharArray()){
            set.add(c);
        }
        for(char sc : s.toCharArray()){
            char current = sc;
            int count = 0;
            while(count < index ) {
                current ++;
                if(current > 'z') current = 'a';
                if(!set.contains(current)) count++;
            }
            answer.append(current);

        }
        return answer.toString();
    }
}

 

다시한번 문제 요약과 의사코드의 중요성 즉 설계의 중요성을 알게되었다..!