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

 

문제

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

 

입력

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

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

 

출력

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

 

 

 

예시 입력

teachermode e

예시 출력

1 0 1 2 1 0 1 2 2 1 0

 

 

 

 

 

문제 풀이

 

[첫번째 풀이]

import java.util.Scanner;
  
public class Main {
  public static void main(String[] args){
    Scanner scanner = new Scanner(System.in);
    String input1 = scanner.next();
    String input2 = scanner.next();

    int[] arr1 = new int[input1.length()];
    int[] arr2 = new int[input1.length()];

    int num = 100;
    for (int i=0; i<input1.length(); i++) {
      if (input2.charAt(0) == input1.charAt(i)) {
        num = 0;
      } else {
        num = num + 1;
      }
      arr1[i] = num;
    }

    num = 100;
    for (int i=input1.length()-1; i>=0; i--) {
      if (input2.charAt(0) == input1.charAt(i)) {
        num = 0;
      } else {
        num = num + 1;
      }
      arr2[i] = num;
    }

    for (int i=0; i<input1.length(); i++) {
      if (arr1[i] <= arr2[i]) {
        System.out.print(arr1[i]);
      } else {
        System.out.print(arr2[i]);
      }
      System.out.print(" ");
    }
  }
}

(Time 147ms / Memory 27MB)

import java.util.Scanner;
  
public class Main {
  public void solution(String input1, String input2) {
    int[] arr1 = new int[input1.length()];
    int[] arr2 = new int[input1.length()];

    int num = 100;
    for (int i=0; i<input1.length(); i++) {
      if (input2.charAt(0) == input1.charAt(i)) {
        num = 0;
      } else {
        num = num + 1;
      }
      arr1[i] = num;
    }

    num = 100;
    for (int i=input1.length()-1; i>=0; i--) {
      if (input2.charAt(0) == input1.charAt(i)) {
        num = 0;
      } else {
        num = num + 1;
      }
      arr2[i] = num;
    }

    for (int i=0; i<input1.length(); i++) {
      if (arr1[i] <= arr2[i]) {
        System.out.print(arr1[i]);
      } else {
        System.out.print(arr2[i]);
      }
      System.out.print(" ");
    }
  }
  
  public static void main(String[] args){
    Scanner scanner = new Scanner(System.in);
    String input1 = scanner.next();
    String input2 = scanner.next();

    Main main = new Main();
    main.solution(input1, input2);
  }
}

(Time 158ms / Memory 27MB)

 

[두번째 풀이]

import java.util.Scanner;
  
public class Main {
  public static void main(String[] args){
    Scanner scanner = new Scanner(System.in);
    String input1 = scanner.next();
    String input2 = scanner.next();

    int[] arr = new int[input1.length()];

    int num = 100;
    for (int i=input1.length()-1; i>=0; i--) {
      if (input2.charAt(0) == input1.charAt(i)) {
        num = 0;
      } else {
        num = num + 1;
      }
      arr[i] = num;
    }

    num = 100;
    for (int i=0; i<input1.length(); i++) {
      if (input2.charAt(0) == input1.charAt(i)) {
        num = 0;
      } else {
        num = num + 1;
      }

      if (num < arr[i]) {
        arr[i] = num;
      }

      System.out.print(arr[i] + " ");
    }
  }
}

(Time 154ms / Memory 27MB)

import java.util.Scanner;
  
public class Main {
  public void solution(String input1, String input2) {
    int[] arr = new int[input1.length()];

    int num = 100;
    for (int i=input1.length()-1; i>=0; i--) {
      if (input2.charAt(0) == input1.charAt(i)) {
        num = 0;
      } else {
        num = num + 1;
      }
      arr[i] = num;
    }

    num = 100;
    for (int i=0; i<input1.length(); i++) {
      if (input2.charAt(0) == input1.charAt(i)) {
        num = 0;
      } else {
        num = num + 1;
      }

      if (num < arr[i]) {
        arr[i] = num;
      }

      System.out.print(arr[i] + " ");
    }
  }
  
  public static void main(String[] args){
    Scanner scanner = new Scanner(System.in);
    String input1 = scanner.next();
    String input2 = scanner.next();

    Main main = new Main();
    main.solution(input1, input2);
  }
}

(Time 150ms / Memory 27MB)

 

 

 

- 혼자 짜보려니 잘 안돼서 풀이 방법 보고 첫번째 풀이처럼 풀었다. for 문이 세 개씩이나 되어 너무 비효율적이라 느껴진다.

- 풀이 영상 답변을 보니 for 문을 두 번만에 풀 수 있길래 따라서 for 문을 두 개로 추려서 두번째 풀이처럼 풀어보았다.