[PS] 인프런 강의 - 문자열 10. 가장 짧은 문자거리

문제

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

 

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

 

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

 

▣ 입력예제 1
teachermode e

 

▣ 출력예제 1
1 0 1 2 1 0 1 2 2 1 0

 

풀이

이번 문제는 함수를 딱히 사용하지 않고 원시적인 방법(?)으로 해결해야 한다.

 

우선 문자열을 char 배열에 저장한뒤 처음부터 쭉 순회하면서 answer 배열에, 왼쪽의 e와의 거리를 저장한다.

 

변수 p을 큰값(1000)으로 초기화해두고, e를 발견하면 p에 0을 저장하고 그 값을 answer[ i ]에 저장한다. 다음 e를 발견할 때까지 p값을 +1시키면서 answer에 값을 저장한다.

 

반복문이 끝나면 answer에 저장되는 값은 이렇게 된다.

 

 

	0	1	2	3	4	5	6	7	8	9	10
s	t	e	a	c	h	e	r	m	o	d	e
answer(1)	1001	0	1	2	3	0	1	2	3	4	0

 

 

왼쪽에 있는 e를 기준으로 떨어진 거리를 저장했으므로,

다시 문자열을 거꾸로 순회해서 오른쪽에 있는 e와의 거리가 더 가까우면 그 거리로 바꿔줘야한다.

 

다시 p를 큰값(1000)으로 초기화하고, 기존 값 answer[ i ] 과 비교해서 작은 값을 answer[ i ]에 저장한다.

 

최종적으로 저장되는 answer는 아래 표와 같다.

 

 

	0	1	2	3	4	5	6	7	8	9	10
s	t	e	a	c	h	e	r	m	o	d	e
answer(2)	1	0	1	2	1	0	1	2	2	1	0

 

코드

import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);

        String s = in.next();
        char t = in.next().charAt(0);
        int[] answer = new int[s.length()];

        int p = 1000;
        for(int i=0; i<s.length()-1; i++) {
            if(s.charAt(i) == t) p = 0;
            else {
                p++;
                answer[i] = p;
            }
        }

        for(int i=s.length()-1; i>=0; i--){
            if(s.charAt(i) == t) p = 0;
            else {
                p++;
                answer[i] = Math.min(answer[i], p);
            }
        }

        for(int x : answer) {
            System.out.print(x + " ");
        }
    }
}