Algorithm/문제

10. 가장 짧은 문자거리

챛채 2023. 5. 24. 16:36

p=1000으로 잡는 이유는 첫 글자처럼 왼쪽 or 오른쪽에 타겟문자 없을 경우 거리를 큰 숫자로 해야 반대 방향의 거리가 작은 값이 되어 선택되기 때문이다.

import java.util.ArrayList;
import java.util.Scanner;

public class Main {
   public int[] solution(String s, char t) {
       int [] answer = new int[s.length()];//s의 길이(0~10)만큼 동적으로 int 배열 만듦
       int p=1000;
       //왼쪽으로부터 떨어진 거리
       for (int i = 0; i < s.length(); i++) {
           if(s.charAt(i)==t){ //i가 t문자(타겟)가 맞으면
               p=0;
               answer[i]=p;
           }
           else{
               p++;
               answer[i]=p;
           }
       }
       //오른쪽부터 떨어진 거리와 기존 값 비교하여 더 작은 숫자로 교체
       p=1000;
       for (int i = s.length()-1; i>=0 ; i--) {
           if(s.charAt(i)==t) //i가 t문자(타겟)가 맞으면
               p=0;
           else{
               p++;
               answer[i]=Math.min(answer[i], p); //기존 값과 비교해서 작은 숫자 고르기
           }
       }

       return answer;
   }
   public static void main(String[] args){
       Main T = new Main();
       Scanner kb = new Scanner(System.in);
       String str = kb.nextLine(); //문자열
       char c = kb.next().charAt(0); //문자 한 개
       for (int x : T.solution(str,c)) {
           System.out.println(x+ " ");
       }
   }
}