(Java) 프로그래머스 - 둘만의 암호

문제 해결 로직은 쉬웠지만 깔끔하게 풀지는 못한 것 같다.

스트림을 이용하면 조금 더 보기 좋게 코드를 작성할 수 있을 것 같은데

아직 스트림이 익숙하지 않다..

---

로직

제시된 String을 char 배열로 바꾸고 contains() 메서드를 이용하기 위해 skipArr을 list에 넣어준다.

arr 배열을 순회하며 문자를 다음 문자로 바꾸어 'z' 이후 문자인지 확인하여 이후 문자라면 'a'로 바꿔준다.

그리고 list에 넘어가야할 문자가 있는지 확인한다.

바꾼 문자가 list에 있다면 cnt를 증가시키지 않고 넘어가고, 없다면 cnt를 증가시켜 cnt가 index와

같아질 때까지 문자를 바꾼다. 마지막으로 바꾼 문자를 StringBuilder에 추가하면 된다.

 

최종 코드

import java.util.ArrayList;
import java.util.List;

class Solution {
    public String solution(String s, String skip, int index) {
        String answer = "";
        StringBuilder stringBuilder = new StringBuilder();
        char[] arr = s.toCharArray();
        char[] skipArr = skip.toCharArray();

        List<Character> list = new ArrayList<>();
        for (char c : skipArr) {
            list.add(c);
        }

        for (int i = 0; i < arr.length; i++) {
            char ch = arr[i];
            int cnt = 0;
            while (cnt < index) {
                ch++;
                if(ch > 'z') {
                    ch = 'a';
                }
                if (list.contains(ch)) {
                    continue;
                }
                cnt++;
            }
            stringBuilder.append(ch);
        }

        answer = stringBuilder.toString();
        return answer;
    }
}