본문 바로가기

Algorithm/Programers - Java

[프로그래머스(Java)] 접미사 배열 / substring

 

Level. 0

 

문제

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

* 제한사항
- my_string은 알파벳 소문자로만 이루어져 있습니다.
- 1 ≤ my_string의 길이 ≤ 100

 

풀이1 

import java.util.Arrays;
class Solution {
    public String[] solution(String my_string) {
        
        String[] answer = new String[my_string.length()];
        String str = "";
        for(int i=answer.length-1; i>=0; i--){
            str = my_string.charAt(i) + str;
            answer[i] = str;
        }
        Arrays.sort(answer);
        return answer;
    }
}

 

풀이2 - stream

import java.util.Arrays;
import java.util.stream.*;
class Solution {
    public String[] solution(String my_string) {

        return IntStream.range(0, my_string.length())
                        .mapToObj(i -> my_string.substring(i)) // mapToObj(myString::substring(i))
                        .sorted()
                        .toArray(String[]::new);

    }
}

 

문자열를 반복하여 생성하는 풀이1 보다 stream을 사용한 풀이2가 실행시간이 더 조금 걸렸다. 

 

 


https://school.programmers.co.kr/learn/courses/30/lessons/181909

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr