[Inflearn] 자바 알고리즘 문제풀이 #01-10 10. 가장 짧은 문자거리

문제

설명

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

 

입력 

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

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

 

출력 

 

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

 

 

 

 

내 풀이

import java.util.Scanner;

public class example {
    public int[] Solution(String s, char c) {
        int[] answer = new int[s.length()];
        char[] arr = s.toCharArray();
        int p = 0;
        for (int i = 0; i < arr.length; i++) {
            if(arr[i] == c) { // 입력받은 문자라면 0
                p=0;
                answer[i] = p;
            }
            else { // 아니라면 +1해서 저장
                p++;
                answer[i] = p;
            }
        }

        p = 0;
        for (int i = arr.length-1; i >= 0; i--) {
            if(arr[i] == c) p=0;
            else {
                p++;
                if(answer[i] > p) answer[i] = p; // 이전 결과보다 p가 작으면 저장
            }
        }

        return answer;
    }

    public static void main(String[] args) {
        example T = new example();
        Scanner sc = new Scanner(System.in);
        String str = sc.next();
        char ch = sc.next().charAt(0);
        for(int x : T.Solution(str, ch)) {
            System.out.print(x + " ");
        }
    }
}

 

 

 

선생님 풀이

import java.util.Scanner;

public class Main1_10 {
    public int[] Solution(String s, char c) {
        int[] answer = new int[s.length()];
        int p = 1000;
        for (int i = 0; i < s.length(); i++) {
            if(s.charAt(i) == c) { // 입력받은 문자라면 0
                p=0;
                answer[i] = p;
            }
            else { // 아니라면 +1해서 저장
                p++;
                answer[i] = p;
            }
        }

        p = 1000;
        for (int i = s.length()-1; i >= 0; i--) {
            if(s.charAt(i) == c) p=0;
            else {
                p++;
                answer[i] = Math.min(answer[i], p);
            }
        }

        return answer;
    }

    public static void main(String[] args) {
        Main1_10 T = new Main1_10();
        Scanner sc = new Scanner(System.in);
        String str = sc.next();
        char ch = sc.next().charAt(0);
        for(int x : T.Solution(str, ch)) {
            System.out.print(x + " ");
        }
    }
}

 

 

 

결과

 

 

 

알게된 점

선생님 풀이 방식에 거의 근접했지만 오른쪽 방향, 왼쪽 방향으로 for문을 두 번 사용할 생각을 못하고

한 큐에 쌍방으로 바꿀 순 없을까? 생각이 계속 들었다. 내 생각보다 컴퓨터는 멍청하기 때문에 단방향으로 먼저 생각하면 좋을 것 같다.

 

그리고 나처럼 굳이 if문으로 비교해서 더 작은 값을 배열에 넣지 않아도, Math.min()을 사용하면 깔끔하다는 점.

작게 감탄했다! 아직 갈 길이 멀다ㅎ__ㅎ 이렇게 적어 놓으면 언젠간 기억하고 써먹겠지..