[JAVA/코딩테스트] 가장 짧은 문자거리

문제

 

import java.util.*;

public class Main {

	public int[] solution(String s, char t){
		int[] answer = new int[s.length()];
		int p = 1000;
		// 왼쪽 문자 t에서 부터의 거리를 저장
		for(int i=0; i<s.length(); i++){
			if(s.charAt(i) == t){
				p = 0;
				answer[i] = p;
			}else{
				p ++;
				answer[i] = p;
			}
		}
		p = 1000;
		// 오른쪽 문자 t에서 부터의 거리를 저장
		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);
			}
		}
		
		return answer;
	}
	
	
    public static void main(String[] args) {
        Main main = new Main();
		Scanner in = new Scanner(System.in);
		String str = in.next();
		char c = in.next().charAt(0);
		for(int x : main.solution(str, c)){
			System.out.print(x + " ");
		}
    }

}