[백준] 알파벳 찾기

문제

알파벳 소문자로만 이루어진 단어 S가 주어진다. 각각의 알파벳에 대해서, 단어에 포함되어 있는 경우에는 처음 등장하는 위치를, 포함되어 있지 않은 경우에는 -1을 출력하는 프로그램을 작성하시오.

 

 

 

입력

첫째 줄에 단어 S가 주어진다. 단어의 길이는 100을 넘지 않으며, 알파벳 소문자로만 이루어져 있다.

 

 

 

출력

각각의 알파벳에 대해서, a가 처음 등장하는 위치, b가 처음 등장하는 위치, ... z가 처음 등장하는 위치를 공백으로 구분해서 출력한다.

만약, 어떤 알파벳이 단어에 포함되어 있지 않다면 -1을 출력한다. 단어의 첫 번째 글자는 0번째 위치이고, 두 번째 글자는 1번째 위치이다.

 

 

 

문제 풀이

나는 맨 처음 문제 접근을 다음과 같이 했다.

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

public class Main {

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        String s = in.nextLine(); // 단어 s
        char[] arr = s.toCharArray(); // 단어 s의 배열
        ArrayList<Integer> answer = new ArrayList<Integer>(); // 답을 저장할 리스트

        for (int i = 0; i < arr.length; i++) {
            for (char j = 'a'; j <= 'z'; j++) {
                if (arr[i] == j) {
                    answer.add(9);
                }
                else {
                    //???
                }
            }
        }

        System.out.println(answer);

    }
}

각각 arr와 a~z까지 for문을 이용해 순회하면서 동일한 알파벳이 있다면 테스트로 9를 출력하고, 아닐 시에는 다른 코드를 실행하려고 했다.

 

하지만 else문에서 어떤 방식을 작성해야할지 몰라서 고민하고 있었다.

결국 아래와 같이 작성했다.

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        String s = in.nextLine(); // 단어 s 입력 받기

        // a부터 z까지 각 알파벳의 첫 등장 위치 찾기
        for (char c = 'a'; c <= 'z'; c++) {
            System.out.print(s.indexOf(c) + " "); // indexOf는 문자가 없으면 -1 반환
        }
    }
}

indexOf()라는 메서드를 이용해 매우 간단하게 푼 것을 확인할 수 있다.

indexOf()라는 메서드가 궁금해서 찾아보았다.

 

indexOf() 메서드는 문자열에서 특정 문자나 부분 문자열이 처음으로 등장하는 위치(인덱스)를 찾아주는 Java의 String 클래스 메서드이다.

 

예를 들어 보겠다.

문자:    s  t  r  a  w  b  e  r  r  y
인덱스:  0  1  2  3  4  5  6  7  8  9

이렇게 있을 때, 결과값을 보면 다음과 같다.

 

"strawberry".indexOf('s') → 0
"strawberry".indexOf('r') → 2 (첫 번째 'r'의 위치)
"strawberry".indexOf('r', 3) → 7 (인덱스 3부터 시작해서 찾으면 두 번째 'r'의 위치)
"strawberry".indexOf('z') → -1 ('z'는 단어에 없음)

 

사실 이 문제는 indexOf() 메서드를 사용하면 바로 해결될 문제였던 것이다.
나는 이 메서드에 대해 모르니 이중 for문을 사용하여 접근하고 있었다.

 

더 발전시켜서 BufferedReader를 사용하는 코드로 바꾸어보았다.

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String s = br.readLine(); // 단어 s 입력 받기

        // a부터 z까지 각 알파벳의 첫 등장 위치 찾기
        for (char c = 'a'; c <= 'z'; c++) {
            System.out.print(s.indexOf(c) + " "); // indexOf는 문자가 없으면 -1 반환
        }

        br.close(); // 사용 후 BufferedReader 닫기
    }
}

최종 완성된 코드는 다음과 같다.

 

이번 문제는 내가 indexOf()라는 메서드를 몰라 헤맸다.
확실히 지원되는 메서드들을 많이 알면 알 수록 문제의 접근법부터 달라져 편하게 풀 수 있는 것 같다.