[인프런] 01-10 가장 짧은 문자거리

문제

---

1.  한 개의 문자열 s와 문자 t가 주어진다.
2.  s의 각 문자가 문자 t와 떨어진 최소거리를 출력한다.

예시

이해

---

• 문자열 s와 t는 공백으로 구분된다.
• 거리 입력을 위한 cnt=100 / 거리 값 누적을 위한 배열 answer
  • s(i) == t 일 때는 0을 누적
  • s(i)!= t 일 때는 cnt++ 후 answer [i] = cnt
  • 이 방법을 두 번 왼쪽에서부터 오른쪽 오른쪽에서 부터 왼쪽 2번 해주면 된다.
• 최소 거리이므로 왼쪽에서 부터 오른쪽에서 비교한 값이랑
  오른쪽에서부터 왼쪽으로 비교한 값이 다르므로 어떻게 해결해야 할지 생각하기

왼쪽에서 부터 오른쪽으로 문자열s의 각 문자가 t와 떨어진 최소 거리
0	1	2	3	4	5	6	7	8	9
t	e	a	c	h	e	r	m	o	d
101	0	1	2	3	0	1	2	3	4

오른쪽에서 부터 왼쪽으로 문자열s의 각 문자가 t와 떨어진 최소 거리
9	8	7	6	5	4	3	2	1	0
d	o	m	r	e	h	c	a	e	t
101	102	103	104	0	1	2	3	0	1

 

• 위 두 개 표를 비교해서 각 번호마다 더 낮은 값을 넣어서 완성시켜 주면 된다.
• 출력할 때 배열이 아닌 공백을 기준으로 출력 되므로 아래와 같이 만들고 return 해주면 된다.

  String.join(" ", answer);​

 

⚠️ 주의사항

---

• 최소거리를 구해야 하니 양쪽에서 출발하는 기준으로 해야 한다.
• 값을 누적시키다가 동일한 값이 나오는 경우 cnt를 초기화시켜준다.
• 출력할 때 배열이 아닌 공백으로 출력해야 한다.

 

문제풀이

---

1. 내가 직접 풀어본 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Java01_10 {

   public String solution(String s, char t) {

        String[] answer = new String[s.length()]; // 빈 배열 등록
        int cnt = 100; // 문자열의 길이 제한 생각하면서 넣기

        for (int i = 0; i < s.length(); i++) {
            if (s.charAt(i) == t) { // i번째 문자열이 입력한 문자 t와 같은지 확인
                answer[i] = String.valueOf(0); // 같으면 0 입력 후 0으로 초기화
                cnt = 0;
            } else {
                cnt ++;
                answer[i] = String.valueOf(cnt);
            }
        }

        for (int i = s.length() - 1; i >= 0; i--) {
            cnt++;

            if (s.charAt(i) == t) {
                cnt = 0;
            } else {
                if (Integer.parseInt(answer[i]) > cnt) { // 앞에 입력 된 배열의 i 번째 값과 현재 입력할려는 cnt 값 비교 후 더 작은 수 입력
                    answer[i] = String.valueOf(cnt);
                }
            }
        }

        String answerToString = String.join(" ", answer); // 출력 형태를 배열 형태가 아닌 칸마다 공백으로 구분한다.

        return answerToString;
    }

    public static void main(String[] args) throws IOException {

        Java01_10 T = new Java01_10();

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");

        String s = st.nextToken();
        char t = st.nextToken().charAt(0);

        System.out.println(T.solution(s,t));
    }
}

---

2. Scanner 방식의 코드

import java.util.*;
class Main {	
	public int[] solution(String s, char t){
		int[] answer=new int[s.length()];
		int p=1000;
		for(int i=0; i<s.length(); i++){
			if(s.charAt(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) 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.next();
		char c=kb.next().charAt(0);
		for(int x : T.solution(str, c)){
			System.out.print(x+" ");
		}
	}
}