프로그래머스 코딩테스트 문제 - [Lv.1] 둘만의 암호 (Java)

정진희·2025년 7월 9일
post-thumbnail

📌 문제

문제 출처 - 링크

문제 설명

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

입출력 예

sskipindexresult
"aukks""wbqd"5"happy"

알고리즘 분류

  • 구현
  • 문자열

💡 알고리즘 설계 / 접근 방법

  1. 문자열 s를 순회한다.

  2. 각 알파벳을 while 문으로 index만큼 순회하며 알파벳을 변환한다.

    • 알파벳을 다음 알파벳으로 바꾼다.

    • 알파벳이 z라면 a로 바꿔준다.

    • 바꾼 알파벳이 skip에 있는지 확인하고, while 문에서 움직일 횟수를 조정한다.


➕ 보완하기 / 성능 비교

  • char 자료형은 ++를 해주면 다음 알파벳으로 넘어갈 수 있다.
    • 해당 문법을 사용하면 빨리 풀 수 있었을텐데 오랜만이라 잊어버려서 char와 아스키코드를 사용해 문제를 풀려다보니 어렵게 풀었고, 테스트 문제도 일부만 통과했다.

✅ 풀이

시간 복잡도 → O(N × M)

  1. for문 : O(N)
  2. while 문
    • 내부 contains : O(M) → M = skip.length()
    • 최대 반복 : O(26) = O(1) (고정 상수)
class Solution {
    public String solution(String s, String skip, int index) {
        StringBuilder sb = new StringBuilder();
        
        // 문자열 s 순회
        for (char ch : s.toCharArray()) {
            int count = 0; // 변환 횟수 기록하기 위한 count
            
            // index 크기 만큼 순회하며 변환
            while (count < index) {
            	// 현재 문자 ch가 z이면 a로 변환, 아니면 다음 문자로 변환
                if (ch == 'z') {
                    ch = 'a';
                } else {
                    ch++;
                }

                // 현재 문자가 skip 문자열에 포함되지 않는 경우
                if (!skip.contains(String.valueOf(ch))) {
                    count++;
                }
            }
            // 변환된 문자 sb에 추가
            sb.append(ch);
        }
        return sb.toString();
    }
}
profile
고민하고, 공부해서 발전하는 개발자가 되자🔥

0개의 댓글