알고리즘

[03.문자열탐색] 가장 짧은 문자거리

devSoo 2022. 3. 27. 21:54

문제

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

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

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

입력예제 1 teachermode e

출력예제 1 10121012210

 

결과

풀었으나 오랜 시간 들임

내가 푼 방식 

const solution = (word, spell) => {
  let arr = [];
  let answer = Array.from({ length: word.length }, () => 0);
  for(let i = 0; i < word.length; i++) {
    if(word[i] === spell) {
      arr.push(i)
    }
  }

  for(let i = 0; i < word.length; i++) {
    let min;
    for(let j=0; j < arr.length; j++) {
	  if(j===0) {
       min = Math.abs(arr[j]-i)
       }
      else {
       if(Math.abs(arr[j]-i) < min) {
          min = Math.abs(arr[j]-i)
        }
      }
    }
    answer[i] = min;
  }
  return answer;
}

접근과정

1.  arr이라는 빈 배열을 만들고 반복문을 돌려 두 번째 인자로 받는 spell과 같은 요소의 인덱스 값을 넣는다

2. 리턴할 answer 배열을 만들고 그 요소로 첫 번째 인자로 받는 str의 length 길이 만큼 0으로 할당한다.

3. str 속 arr 을 이중 반복문을 돌린다.

4. 이중 반복문 속 두 번째 arr 반복문에서 조건문을 통해 첫 번째 인덱스일 경우 최소값 min을  arr의 value 값과 첫 번째 반복문의 인덱스 값을 뺀 것의 절대값(거리)을 넣어준다. 

5. 그 외의 경우 같은 방식으로 거리를 구해주되 아까 설정한 min과의 비교를 통해 더 작은 값이 min이 되게 설정한다. 

6. 두 번째 반복문이 끝났을 때 제일 작은 min값을 answer[i]의 값으로 할당해주고 그 answer를 리턴한다. 

모범답안

function solution(s, t){
    let answer=[];
    let p=1000;
    for(let x of s){
        if(x===t){
            p=0;
            answer.push(p);
        }
        else{
            p++;
            answer.push(p);
        }
    }
    p=1000;
    for(let i=s.length-1; i>=0; i--){
        if(s[i]===t) p=0;
        else{
            p++;
            answer[i]=Math.min(answer[i], p);
        }
    }
    return answer;
}

리뷰