프로그래머스 코딩테스트 1. 시저 암호(JavaScript)

문제설명

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

제한 조건

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

입출력 예

코드(실패)

function solution(s, n) {
    let answer = '';
    for(let i = 0; i < s.length; i++) {
        let code; 
        if(s[i] === " ") {
            answer += " ";
          // i++해주는 것이 연속으로 " "가 들어오는 경우 잡지 못하게 됨.
            i++;
        }
        code = s.charCodeAt(i);
        if(code >= 65 && code <= 90) {
            if(code + n > 90) {
                answer += String.fromCharCode(code + n - 26);
            } else {
                answer += String.fromCharCode(code+n);
            }
        }
        if(code >= 97 && code <= 122) {
            if(code + n > 122) {
                answer += String.fromCharCode(code + n - 26);
            } else {
                answer += String.fromCharCode(code+n);
            }
        }
    }
    return answer;
}

테스트 9, 11, 12, 13 에서 실패가 떴다. 이유가 무엇일까 . . . ?
테스트 케이스를 추가하면서 살펴보니. 공백이 연속으로 들어오는 경우 실패하는 것을 발견하였다.

코드(통과)

function solution(s, n) {
    let answer = '';
    for(let i = 0; i < s.length; i++) {
        let code; 
      // s[i]가 " "인 경우 answer에 " "문자열을 더해주었다.
        if(s[i] === " ") {
            answer += " ";
        }
      // code라는 변수에 해당 아스키코드값을 넣어준다.
        code = s.charCodeAt(i);
      // 아스키코드 번호가 이에 해당될 때 
        if(code >= 65 && code <= 90) {
            if(code + n > 90) {
              // String.fromCharCode를 통해 해당하는 아스키코드 값의 문자를 answer에 추가해 준다.
              // 이 때 대문자의 경우는 90을 넘으면 대문자가 아닌 다른 문자이므로
              // -26을 해주어서 Z 다음에 A부터 시작하도록 하였다.
                answer += String.fromCharCode(code + n - 26);
            } else {
                answer += String.fromCharCode(code+n);
            }
        }
      // 위와 마찬가지로 ! 이것은 소문자 아스키코드 값 해당하면
        if(code >= 97 && code <= 122) {
          // 아스키 코드의 값이 122가 넘어가면 소문자가 아닌 다른 문자이기 때문에
            if(code + n > 122) {
              // -26을 해주어서 z 다음에 a가 되도록 해준다.
                answer += String.fromCharCode(code + n - 26);
            } else {
                answer += String.fromCharCode(code+n);
            }
        }
    }
    return answer;
}