가장 짧은 문자거리

설명

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

입력

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

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

출력

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

 

import java.util.Scanner;

class Main {

    public int[] soulution(String st, char c) {
        int[] answer = new int[st.length()];
        int p = 1000;
        for (int i = 0; i < st.length(); i++) {
            if (st.charAt(i) == c) {
                answer[i] = 0;
                p=0;
            } else {
                p++;
                answer[i] = p;
            }
        }
        p=1000;
        for (int i=st.length()-1; i>=0; i--) {
            if (st.charAt(i) == c) {
                p=0;
            } else {
                p++;
                answer[i] = Math.min(answer[i], p);
            }
        }


        return answer;
    }

    public static void main(String[] args) {
        Main main = new Main();
        Scanner sc = new Scanner(System.in);
        String input = sc.next();
        char c = sc.next().charAt(0);
        for (int a : main.soulution(input, c)) {
            System.out.print(a + " ");
        }
    }
}