[JAVA] 알고리즘 문제풀이 입문 1.10번

강의: https://inf.run/w779

자바(Java) 알고리즘 문제풀이 입문: 코딩테스트 대비 - 인프런 | 강의

 

1.10번

import java.util.Scanner;

 

public class Main {

  public int[] solution(String str, char x){

    int[] answer = new int[str.length()];

    int num = 1000;

    for(int i = 0; i < str.length(); i++){

      if(str.charAt(i) == x){

        num = 0;

        answer[i] = num;

      }else{

        num++;

        answer[i] = num;

      }

    }

    for(int i = str.length()-1; i >= 0; i--){

      if(str.charAt(i) == x){

        num = 0;

        answer[i] = num;

      }else{

        num++;

        if(answer[i] > num)

        answer[i] = num;

      }

    }

    return answer;

  }

 

  public static void main(String[] args){

    Main m = new Main();

    Scanner in=new Scanner(System.in);

    String input = in.next();

    char x = in.next().charAt(0);

    in.close();

    for(int n : m.solution(input, x)){

      System.out.print(n+" ");

    }

  }

}