[Baekjoon 11656] 접미사 배열 - JAVA

[Silver IV] 접미사 배열 - 11656

문제 링크

성능 요약

메모리: 17812 KB, 시간: 168 ms

분류

정렬, 문자열

문제 설명

접미사 배열은 문자열 S의 모든 접미사를 사전순으로 정렬해 놓은 배열이다.

baekjoon의 접미사는 baekjoon, aekjoon, ekjoon, kjoon, joon, oon, on, n 으로 총 8가지가 있고, 이를 사전순으로 정렬하면, aekjoon, baekjoon, ekjoon, joon, kjoon, n, on, oon이 된다.

문자열 S가 주어졌을 때, 모든 접미사를 사전순으로 정렬한 다음 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 문자열 S가 주어진다. S는 알파벳 소문자로만 이루어져 있고, 길이는 1,000보다 작거나 같다.

출력

첫째 줄부터 S의 접미사를 사전순으로 한 줄에 하나씩 출력한다.

문제 풀이

그냥 입력받은 문자열을 배열에 쪼개서 넣고 정렬시키면 끝난다.

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();

        String word = br.readLine();

        String[] suffixes = new String[word.length()];
        for (int i = 0; i < suffixes.length; i++) {
            suffixes[i] = word.substring(i);
        }
        Arrays.sort(suffixes);

        for (String suffix : suffixes) {
            sb.append(suffix).append("\n");
        }

        System.out.print(sb);
    }
}

코드 풀이

String word = reader.readLine();

String[] suffixes = new String[word.length()];
for (int i = 0; i < suffixes.length; i++) {
    suffixes[i] = word.substring(i);
}

• 입력받은 단어를 word에 저장한다.
• 입력된 단어의 길이만큼의 크기를 가지는 배열 suffixes를 생성한다.
• 각 인덱스에 해당하는 접미사를 substring 메서드를 사용하여 생성한다.

        Arrays.sort(suffixes);

        for (String suffix : suffixes) {
            sb.append(suffix).append("\n");
        }

        System.out.print(sb);
}

• Arrays 클래스의 sort 메서드를 사용하여 suffixes 배열을 사전순으로 정렬한다.
• 정렬된 suffixes 배열의 각 원소를 순회하며, result에 각 접미사를 추가한다.
• 접미사 사이에 개행 문자(\n)를 추가하여 한 줄에 하나씩 출력한다.
• 최종적으로 정렬된 접미사를 출력힌다.