[프로그래머스] LV.1 시저 암호 - 자바 [38/80]

💬 문제 설명

어떤 문장의 각 알파벳을 일정한 거리만큼 밀어서 다른 알파벳으로 바꾸는 암호화 방식을 시저 암호라고 합니다. 예를 들어 "AB"는 1만큼 밀면 "BC"가 되고, 3만큼 밀면 "DE"가 됩니다. "z"는 1만큼 밀면 "a"가 됩니다. 문자열 s와 거리 n을 입력받아 s를 n만큼 민 암호문을 만드는 함수, solution을 완성해 보세요.

 

🚫 제한 조건

• 공백은 아무리 밀어도 공백입니다.
• s는 알파벳 소문자, 대문자, 공백으로만 이루어져 있습니다.
• s의 길이는 8000이하입니다.
• n은 1 이상, 25이하인 자연수입니다.

 

📢 입출력 예

 

📃 제출 코드

import java.util.stream.Collectors;

class Solution {
    public String solution(String s, int n) {
        String answer = 
            s.chars()
            .mapToObj(ch -> (char) ch)
            .map(ch -> {
                if (Character.isLowerCase(ch)) {
                    return (char) ((ch - 'a' + n) % 26 + 'a');
                } else if (Character.isUpperCase(ch)) {
                    return (char) ((ch - 'A' + n) % 26 + 'A'); 
                } return ch;
            })
            .map(String::valueOf)
            .collect(Collectors.joining());
        
        return answer;
    }
}

 

✏ 해결방법 & 배운점

이 문제는 다른 방법으로도 해결 할 수 있었겠지만.. 스트림을 복습하고자 꾸역꾸역 스트림으로 풀어보았습니다.

• s.chars() : 문자열을 IntStream으로 변환한다. 각 int 값은 문자의 ASCII 값을 나타냅니다.
• mapToObj(ch -> (char) ch)을 사용하여 문자의  ASCII 값을 문자로 변경
• map(ch -> { ... })에서는 각 문자에 대해 이동 처리합니다.
  • 소문자: ((ch - 'a' + n) % 26 + 'a')를 사용하여 알파벳 순환 처리합니다.
  • 대문자: ((ch - 'A' + n) % 26 + 'A')를 사용하여 대문자의 알파벳 순환 처리를 수행합니다.
  • 알파벳이 아닌 문자는 그대로 반환: 알파벳이 아닌 문자는 그대로 유지합니다.
• map(String::valueOf)을 사용하여 각 문자를 다시 문자열로 변환합니다.
• collect(Collectors.joining())을 사용하여 변환된 문자열을 하나의 문자열로 합칩니다.

 

import java.util.stream.Collectors;

class Solution {
    public String solution(String str, int n) {
        String collect = str.chars()
                .map(ch -> {
                    if (Character.isLowerCase(ch)) {
                        return (ch - 'a' + n) % 26 + 'a'; // 소문자 알파벳 순환 처리
                    } else if (Character.isUpperCase(ch)) {
                        return (ch - 'A' + n) % 26 + 'A'; // 대문자 알파벳 순환 처리
                    }
                    return ch; // 알파벳이 아닌 문자는 그대로 반환
                })
                .mapToObj(ch -> String.valueOf((char) ch)) // ASCII 값을 문자로 변환 후 문자열로 변환
                .collect(Collectors.joining()); // 문자열로 합침

        return collect;
    }
}

++ 소스코드를 조금더 줄여 보았습니다.