Algorithm/Programmers 입문

[프로그래머스 코딩기초 트레이닝] 접미사 배열 (JAVA)

JunFe 2023. 11. 6. 23:54
반응형

문제 설명

어떤 문자열에 대해서 접미사는 특정 인덱스부터 시작하는 문자열을 의미합니다. 예를 들어, "banana"의 모든 접미사는 "banana", "anana", "nana", "ana", "na", "a"입니다. 문자열 my_string이 매개변수로 주어질 때, my_string의 모든 접미사를 사전순으로 정렬한 문자열 배열을 return 하는 solution 함수를 작성해 주세요.


제한사항

my_string은 알파벳 소문자로만 이루어져 있습니다.

1 ≤ my_string의 길이 ≤ 100


입출력 예

my_string result
"banana" ["a", "ana", "anana", "banana", "na", "nana"]
"programmers" ["ammers", "ers", "grammers", "mers", "mmers", "ogrammers", "programmers", "rammers", "rogrammers", "rs", "s"]

입출력 예 설명

입출력 예 #1

예제 1번의 my_string는 "banana"로 모든 접미사는 문제의 설명과 같습니다. 이를 사전순으로 정렬하면 "a", "ana", "anana", "banana", "na", "nana"이므로 ["a", "ana", "anana", "banana", "na", "nana"]를 return 합니다.

 

입출력 예 #2

예제 2번의 my_string는 "programmers"이고 모든 접미사는 "programmers", "rogrammers", "ogrammers", "grammers", "rammers", "ammers", "mmers", "mers", "ers", "rs", "s"입니다. 이를 사전순으로 정렬한 문자열 배열 ["ammers", "ers", "grammers", "mers", "mmers", "ogrammers", "programmers", "rammers", "rogrammers", "rs", "s"]를 return 합니다.


풀이

-> 먼저 temp배열을 만들어서 만들 수 있는 모든 접미사를 만들어서 담아준다. 그 후에 사전적 배열로 어떻게 만들지 고민을 많이 해봤는데 sort함수를 사용하니 간단하게 해결되었다. 그냥 사전순 배열은 어떻게 할 수 있을지 고민해보도록 해야겠다.


코드

import java.util.*;
class Solution {
    public String[] solution(String my_string) {
        String[] answer = {};
        int length = my_string.length();
        String[] temp = new String[length];
        for(int i=0; i<length;i++){
            temp[i] = my_string.substring(i, length);
        }
        Arrays.sort(temp); // temp 배열을 사전순으로 정렬
        answer = temp;
        return answer;
    }
}
반응형