문제 설명

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

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

1차 코드

class Solution {
    public String solution(String s, int n) {
        String answer = "";
        
        for(int i = 0; i < s.length(); i++){
            char c = s.charAt(i);
            
            if(c == ' '){
                answer += ' ';
            }
            else if(Character.isUpperCase(c) && c + n > 90){
                answer += (char)(c + n - 26);
            }
            else if(Character.isLowerCase(c) && c + n > 122){
                answer += (char)(c + n - 26);
            }
            else{
                answer += (char)(c + n);
            }
        }  

        return answer;
    }
}

 

조건문 내부에 answer += (c + n);을 그대로 answer에 넣게 되면 문자가 아니라 숫자가 문자열로 들어가기 때문에 (char)로 형변환을 꼭 해주어야 한다.

 

개선된 코드 - StringBuilder 사용

class Solution {
    public String solution(String s, int n) {
        String answer = "";
        StringBuilder sb = new StringBuilder(answer);
        
        for(int i = 0; i < s.length(); i++){
            char c = s.charAt(i);
            
            if(c == ' '){
                sb.append(' ');
            }
            else if(Character.isUpperCase(c) && c + n > 90){
                sb.append((char)(c + n - 26));
            }
            else if(Character.isLowerCase(c) && c + n > 122){
                sb.append((char)(c + n - 26));
            }
            else{
                sb.append((char)(c + n));
            }
        }  

        return sb.toString();
    }
}

 

String은 불변객체이기 때문에, 기존 코드대로 +를 사용해서 문자열을 붙이면 계속 새로운 객체를 생성해야 한다.

하지만 StringBuilder는 내부적으로 가변 배열(char[])을 사용해서, 문자열을 덧붙일 때마다 같은 객체에 계속 추가한다.

문자열 결합을 한 객체 안에서 처리하기 때문에 메모리도 아낄 수 있고 속도가 훨씬 빨라진다.

티스토리툴바