[Java] 가장 짧은 문자거리

설명

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

입력

첫 번째 줄에 문자열 s와 문자 t가 주어진다. 문자열과 문자는 소문자로만 주어집니다.

문자열의 길이는 100을 넘지 않는다.

출력

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

예시 입력 1 

teachermode e

예시 출력 1

1 0 1 2 1 0 1 2 2 1 0

 

 

풀이

해당 문제는 문자열 word에 각 문자가 문자 target과 떨어진 최소 거리를 구하는 문제이다.

먼저, 왼쪽에서 오른쪽으로 최소 거리를 구했다.

현재 문자와 타켓이 같지 않으면 position++을 증감해주고, 같으면 떨어진 최소 거리는 0이기에 갱신했다.

 

두 번째는 오른쪽에서 왼쪽으로 최소 거리를 구하면서 이전에 배열에 할당했던 값을 비교해서 왼쪽보다 오른쪽의 거리가 더 가까우면 최소 거리를 갱신해서 해결했다.

package section01.E10_가장_짧은_문자거리;

import java.io.*;
import java.util.StringTokenizer;

public class Main {

    public int[] solution(String word, char target){

        int wordLength = word.length();
        int[] answer = new int[wordLength];
        int position = 1000;

        for (int i = 0; i < wordLength; i++){
            if (word.charAt(i) == target){
                position = 0;
                answer[i] = position;
            } else {
                position++;
                answer[i] = position;
            }
        }

        position = 1000;
        for (int i = wordLength-1; i >= 0; i--){
            if (word.charAt(i) == target){
                position = 0;
            } else {
                position++;
                int min = Math.min(answer[i], position);
                answer[i] = min;
            }
        }

        return answer;
    }

    public static void main(String[] args) throws IOException {
        Main T = new Main();
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        StringTokenizer st = new StringTokenizer(br.readLine());

        String word = st.nextToken();
        char findChar = st.nextToken().charAt(0);

        for (int answer : T.solution(word, findChar)){
            bw.write(String.valueOf(answer) + " ");
        }

        bw.flush();
        bw.close();
    }
}