[알고리즘] 문자 거리

문제 설명

한 개의 문자열 s와 문자 t가 주어지면 문자열 s의 각 문자가 문자 t와 떨어진 최소거리를 출 력하는 프로그램을 작성하세요.

입력 설명

첫 번째 줄에 문자열 s와 문자 t가 주어진다. 문자열과 문자는 소문자로만 주어집니다. 문자열의 길이는 100을 넘지 않는다.

출력 설명

첫 번째 줄에 각 문자열 s의 각 문자가 문자 t와 떨어진 거리를 순서대로 출력한다.

입력예제 1

teachermode e

출력예제 1

10121012210

문제 풀이

const s = 'teachermode';
const t = 'e';

solution(s, t);

function solution() {
  let count = 0;
  let answer = [];

  for (let i = 0; i < s.length; i++) {
    if (s[i] !== t) {
      count++;
      answer.push(count);
    } else {
      count = 0;
      answer.push(count);
    }
  }

  for (let i = s.length - 1; i >= 0; i--) {
    if (s[i] !== t) {
      console.log(s[i]);
      count++;
      if (answer[i] > count) {
        answer[i] = count;
      }
    } else {
      count = 0;
    }
  }

  return answer;
}