JavaScript

문자열

문제

상근이의 할머니는 아래 그림과 같이 오래된 다이얼 전화기를 사용한다.

dial

전화를 걸고 싶은 번호가 있다면, 숫자를 하나를 누른 다음에 금속 핀이 있는 곳 까지 시계방향으로 돌려야 한다. 숫자를 하나 누르면 다이얼이 처음 위치로 돌아가고, 다음 숫자를 누르려면 다이얼을 처음 위치에서 다시 돌려야 한다.

숫자 1을 걸려면 총 2초가 필요하다. 1보다 큰 수를 거는데 걸리는 시간은 이보다 더 걸리며, 한 칸 옆에 있는 숫자를 걸기 위해선 1초씩 더 걸린다.

상근이의 할머니는 전화 번호를 각 숫자에 해당하는 문자로 외운다. 즉, 어떤 단어를 걸 때, 각 알파벳에 해당하는 숫자를 걸면 된다. 예를 들어, UNUCIC는 868242와 같다.

할머니가 외운 단어가 주어졌을 때, 이 전화를 걸기 위해서 필요한 최소 시간을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 알파벳 대문자로 이루어진 단어가 주어진다. 단어의 길이는 2보다 크거나 같고, 15보다 작거나 같다.

출력

첫째 줄에 다이얼을 걸기 위해서 필요한 최소 시간을 출력한다.

예제

입력

WA

출력

입력

UNUCIC

출력

내 코드

const input = require('fs').readFileSync('/dev/stdin').toString();

let times = 0;
const defTime = 2;
const nums = {
  2: ['A', 'B', 'C'],
  3: ['D', 'E', 'F'],
  4: ['G', 'H', 'I'],
  5: ['J', 'K', 'L'],
  6: ['M', 'N', 'O'],
  7: ['P', 'Q', 'R', 'S'],
  8: ['T', 'U', 'V'],
  9: ['W', 'X', 'Y', 'Z'],
};

for (let i of input) {
  for (let j = 2; j <= 9; j++) {
    if (nums[j].includes(i)) {
      times += defTime + j - 1;
    }
  }
}

console.log(times);

우선 각 숫자에 해당하는 알파벳 배열을 담은 객체를 선언한다.
input 문자열을 for..of 반복문으로 2부터 9까지 반복한다.

2부터 9까지 해당 알파벳이 있는지 확인한 후, (includes()가 true)
있다면 times에 defTime 2초 + (j-1)초를 더한다.

defTime은 1까지 돌리는데에 2초가 기본으로 소요되므로,
만약 숫자가 3이라면 시작->1->3 까지 총 2+2초가 소요된다.

실행 결과

120ms / 9324KB

리팩토링

이차원 배열로 바꾼 버전.

const input = require('fs').readFileSync('/dev/stdin').toString();

let times = 0;
const defTime = 2;
const nums = [
  ['A', 'B', 'C'],
  ['D', 'E', 'F'],
  ['G', 'H', 'I'],
  ['J', 'K', 'L'],
  ['M', 'N', 'O'],
  ['P', 'Q', 'R', 'S'],
  ['T', 'U', 'V'],
  ['W', 'X', 'Y', 'Z'],
];


for (let i of input) {
  for (let j = 0; j < 8; j++) {
    if (nums[j].includes(i)) {
      times += defTime + j + 1;
    }
  }
}

console.log(times);

참고로 성능은 비슷하다. (120ms)
개인적으로 객체로 표현하는 것이 더 보기 쉬운 것 같다.