[프로그래머스] 시저 암호 (자바스크립트)

 

https://school.programmers.co.kr/learn/courses/30/lessons/12926

프로그래머스

 

문제

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

 

제한 사항

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

입출력 예

s	n	result
"AB"	1	"BC"
"z"	1	"a"
"a B z"	4	"e F d"

 

접근 방법

인덱스로 접근해야 할 테니 일단 대문자 스트링 집합("ABCDE…Z")과 소문자 스트링 집합을 만든다

여기서 문제는 n씩 밀었을 때 z보다 뒤로 밀리는 경우를 고려해야 한다는 것이다. 문자열 전체의 길이는 26이고

A + 26 하면 다시 A로 돌아오게 된다는걸 생각하자. A + 27이면? B가 될 것이다.

즉 수식으로 생각해보면 현재 인덱스 + n >= 26  인 경우 26을 빼주면 된다. 전체 문자열 길이보다 더 많이 밀리는 경우 한바퀴를 빼준다는 느낌으로

그리고 공백인 문자열은 그냥 그대로 반환해주면 된다.

 

해결 코드

function solution(s, n) {
        const big = "ABCDEFGHIJKLMNOPQRSTUVWXYZ".split("");
        const small = "abcdefghijklmnopqrstuvwxyz".split("");

        const arr = s.split("").map((el, i) => {
          if (el !== " ") {
            if (big.includes(el)) {
              if (big.indexOf(el) + n >= 26)
                return (el = big[big.indexOf(el) + n - 26]);
              return (el = big[big.indexOf(el) + n]);
            } else {
              if (small.indexOf(el) + n >= 26)
                return (el = small[small.indexOf(el) + n - 26]);
              return (el = small[small.indexOf(el) + n]);
            }
          } else {
            return (el = " ");
          }
        });
        return arr.join("");
      }

 

보완

if문이 너무 많은데..? 문자 코드와 정규표현식을 쓰면 코드가 훨씬 짧아지더라. 잘 모르는 개념이라 일단은 이렇게 풀었다.

indexOf는 스트링에도 쓸 수 있어서 배열 변환 안해도 됐었다 ...🙄

그리고 제한 사항에

• n은 1 이상, 25이하인 자연수입니다.

이런 조항이 있는데 이런거 잘 읽어야 코드를 재밌게 짤 수 있다. 빨리 푼다고 정신팔려서 n이 100일수도 있잖아 이러고 있었는데 25까지라면 그냥

const big = "ABCDEFGHIJKLMNOPQRSTUVWXYZABCDEFGHIJKLMNOPQRSTUVWXYZ"

이렇게 두번 쓰면 된다. 두바퀴 세바퀴 밀릴 일이 없다는 얘기기 때문에 26을 빼주고 말고 하는 귀찮은 과정이 없어진다.