백준 / 10809번 / 알파벳 찾기

[문제]

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

[입력]

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

[출력]

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

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

[예제 입력]

5
20 10 35 30 7

[예제 출력]

1 0 -1 -1 2 -1 -1 -1 -1 4 3 -1 -1 7 5 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1

 

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

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        String[] arr = { "a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n", "o", "p", "q", "r", "s",
                "t", "u", "v", "w", "x", "y", "z" };
        
        String[] s = sc.next().split("");
        ArrayList<String> list = new ArrayList<>(Arrays.asList(s)); 

        int[] result = new int[arr.length];
        for (int k = 0; k < result.length; k++) {
            result[k] = -1;
        }

        for (int i = 0; i < list.size(); i++) {
            for (int j = 0; j < arr.length; j++) {
                if(list.get(i).equals(arr[j])) {
                    if(result[j] != -1) {
                        continue;   
                    }
                    result[j] = i;
                    break;
                }
            }
        }

        for (int i : result) {
            System.out.print(i + " ");
        }
    }

}

 

ArrayList에 입력받은 문자열을 넣어주고 for문을 돌면서

알파벳이 저장돼있던 arr배열의 값을 입력받은 문자열의 문자 위치 (인덱스)로 바꿔준다.

 

단, 

baekjoon 처럼 o가 중복되는 경우(5,6번 인덱스) 처음 등장한 위치를 저장해줘야 하기 때문에

for문 안에 result[j] != -1인 경우 (이미 값이 한 번 들어온 경우)는 값을 대입하지 않도록 했다.

 

레벨을 보니 간단한 문제 같은데, 생각보다 코드가 좀 길고 복잡한 느낌이 있어 다른 사람들의 풀이도 찾아봤다.

 

---

배열에 알파벳을 넣어 비교하지 않고, 해당 알파벳을 숫자로 변환하는 법

즉, 해당 문자의 알파벳에서 -'a'를 빼 바로 위치를 배열에 저장하는 방법이다.

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        
        String s = sc.next();

        // a,b,c... 알파벳이 등장한 위치 저장!
        int[] result = new int[26];
        for (int k = 0; k < result.length; k++) {
            result[k] = -1;
        }

        // 단어의 특정 알파벳 위치를 result 배열에 저장
        for (int i = 0; i < s.length(); i++) {
            char ch = s.charAt(i); 
            
            if(result[i] != -1) {
                continue;   
            }
            result[ch - 'a'] = i;
        }

        for (int i : result) {
            System.out.print(i + " ");
        }   
    }
}

처음 등장한 위치 저장하는 것은 위의 코드와 같은 방법으로 해줬다.

 

---

 

더 간단하고 쉬운 방법은 String 클래스의 indexOf()를 사용하는 방법이다. 

indexOf 메소드는 특정 문자의 위치를 반환하고, 존재하지 않는 경우 -1을 반환한다.

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        String s = sc.next();
        for(char c='a'; c<='z'; c++) {
            System.out.print(s.indexOf(c) + " ");
        }
    }
}

이렇게 간단하게 for문을 통해 알파벳이 단어의 몇 번째 위치에 존재하는 지 알 수 있다.