[JAVA]String10.가장 짧은 문자 거리

설명

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

입력

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

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

 

출력

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

예시 입력 1	예시 출력 1
teachermode e	1 0 1 2 1 0 1 2 2 1 0

 

풀이

import java.util.Scanner;
class Main {
	
	public int[] solution(String str, char s) {
		
		int[] answer = new int[str.length()];
		
		int p = 1000;
		
		for(int i = 0; i < str.length(); i++) { 
			
			if(str.charAt(i) == s) {
				p =0;
				answer[i] = p;
			} else {
				p++;
				answer[i] = p;
			}
		}
		
		p = 1000;
		
		for(int i = str.length() - 1; i >= 0; i--) {
			if(str.charAt(i) == s) p = 0;
			else {
				p++;
				answer[i]=Math.min(answer[i], p);
			}
		}
	
		return answer;
 
	
}
	public static void main(String[] args){
		Main t = new Main();
		Scanner sc = new Scanner(System.in);
		String str =  sc.nextLine();
		char s = sc.next().charAt(0);
		for(int x : t.solution(str, s)) {
			System.out.print(x + " ");
	}
}

 

 

해설

- str 과 문자 s의 최소 거리 구하기

- 두 거리 중 최소값 Math.min(anwser[i], p);