[프로그래머스] 시저 암호 - JavaScript

function solution(s, n) {
  // 알파벳 대문자 소문자 문자열을 정의해준다.
  const upper = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
  const lower = "abcdefghijklmnopqrstuvwxyz";

  // 알파벳의 길이를 정의해준다.
  const length = upper.length;

  let answer = "";

  // 파라미터로 들어온 s의 크기만큼 for문을 돌려준다.
  for (let i = 0; i < s.length; i++) {
    // includes() 메서드와 삼항 연산자를 통해
    // s[i]가 대문자에 들어있으면 position 변수에 upper을
    // s[i]가 소문자에 들어있으면 position 변수에 lower을
    // 공백일 경우 null을 넣어준다.
    let position = upper.includes(s[i])
      ? upper
      : lower.includes(s[i])
      ? lower
      : null;

    if (position) {
      // indexOf() 메서드를 사용해 해당 문자의 인덱스를 찾고 n을 더해준다.
      const idx = position.indexOf(s[i]) + n;

      // n을 더한 인덱스가 알파벳의 길이보다 클 경우
      // 인덱스에서 알파벳 길이를 뺀 값을 찾아주고
      // 아닐경우 그 인덱스에 해당하는 문자를 answer에 넣어준다.
      answer += idx >= length ? position[idx - length] : position[idx];
    } else {
      // position이 null일 경우 answer에 공백을 넣어준다.
      answer += " ";
    }
  }

  // 최종 값을 리턴한다.
  return answer;
}

console.log(solution("AB", 1)); // "BC"
console.log(solution("z", 1)); // "a"
console.log(solution("a B z", 4)); // "e F d"